[INFO] cloning repository https://github.com/egraphs-good/egglog-experimental
[INFO] running `Command { std: "git" "-c" "credential.helper=" "-c" "credential.helper=/workspace/cargo-home/bin/git-credential-null" "clone" "--bare" "https://github.com/egraphs-good/egglog-experimental" "/workspace/cache/git-repos/https%3A%2F%2Fgithub.com%2Fegraphs-good%2Fegglog-experimental", kill_on_drop: false }`
[INFO] [stderr] Cloning into bare repository '/workspace/cache/git-repos/https%3A%2F%2Fgithub.com%2Fegraphs-good%2Fegglog-experimental'...
[INFO] running `Command { std: "git" "rev-parse" "HEAD", kill_on_drop: false }`
[INFO] [stdout] 2a5085e7a7cff59d2729c1f3f22333944d58d7d0
[INFO] testing egraphs-good/egglog-experimental against beta-2025-01-12 for beta-1.85-1
[INFO] running `Command { std: "git" "clone" "/workspace/cache/git-repos/https%3A%2F%2Fgithub.com%2Fegraphs-good%2Fegglog-experimental" "/workspace/builds/worker-2-tc2/source", kill_on_drop: false }`
[INFO] [stderr] Cloning into '/workspace/builds/worker-2-tc2/source'...
[INFO] [stderr] done.
[INFO] validating manifest of git repo https://github.com/egraphs-good/egglog-experimental on toolchain beta-2025-01-12
[INFO] running `Command { std: CARGO_HOME="/workspace/cargo-home" RUSTUP_HOME="/workspace/rustup-home" "/workspace/cargo-home/bin/cargo" "+beta-2025-01-12" "metadata" "--manifest-path" "Cargo.toml" "--no-deps", kill_on_drop: false }`
[INFO] started tweaking git repo https://github.com/egraphs-good/egglog-experimental
[INFO] removed 0 missing tests
[INFO] finished tweaking git repo https://github.com/egraphs-good/egglog-experimental
[INFO] tweaked toml for git repo https://github.com/egraphs-good/egglog-experimental written to /workspace/builds/worker-2-tc2/source/Cargo.toml
[INFO] crate git repo https://github.com/egraphs-good/egglog-experimental already has a lockfile, it will not be regenerated
[INFO] running `Command { std: CARGO_HOME="/workspace/cargo-home" RUSTUP_HOME="/workspace/rustup-home" "/workspace/cargo-home/bin/cargo" "+beta-2025-01-12" "fetch" "--manifest-path" "Cargo.toml", kill_on_drop: false }`
[INFO] [stderr]     Blocking waiting for file lock on package cache
[INFO] [stderr]     Updating git repository `https://github.com/egraphs-good/egglog.git`
[INFO] [stderr]     Updating crates.io index
[INFO] [stderr]     Blocking waiting for file lock on package cache
[INFO] [stderr]  Downloading crates ...
[INFO] [stderr]   Downloaded codspeed v2.7.2
[INFO] [stderr]   Downloaded libtest-mimic v0.6.1
[INFO] [stderr]   Downloaded codspeed-criterion-compat v2.7.2
[INFO] [stderr]   Downloaded symbol_table v0.4.0
[INFO] [stderr]   Downloaded egraph-serialize v0.2.0
[INFO] running `Command { std: "docker" "create" "-v" "/var/lib/crater-agent-workspace/builds/worker-2-tc2/target:/opt/rustwide/target:rw,Z" "-v" "/var/lib/crater-agent-workspace/builds/worker-2-tc2/source:/opt/rustwide/workdir:ro,Z" "-v" "/var/lib/crater-agent-workspace/cargo-home:/opt/rustwide/cargo-home:ro,Z" "-v" "/var/lib/crater-agent-workspace/rustup-home:/opt/rustwide/rustup-home:ro,Z" "-e" "SOURCE_DIR=/opt/rustwide/workdir" "-e" "CARGO_TARGET_DIR=/opt/rustwide/target" "-e" "CARGO_HOME=/opt/rustwide/cargo-home" "-e" "RUSTUP_HOME=/opt/rustwide/rustup-home" "-w" "/opt/rustwide/workdir" "-m" "1610612736" "--user" "0:0" "--network" "none" "ghcr.io/rust-lang/crates-build-env/linux@sha256:86ea7c7af713d31e8cfdb68a6d0db50b5cf7cbeecde3d112f9f257f747318d36" "/opt/rustwide/cargo-home/bin/cargo" "+beta-2025-01-12" "metadata" "--no-deps" "--format-version=1", kill_on_drop: false }`
[INFO] [stdout] 92abd7de59f6720c4d28d10762c68a8a896896aa44b64c48d7065435961c06d3
[INFO] running `Command { std: "docker" "start" "-a" "92abd7de59f6720c4d28d10762c68a8a896896aa44b64c48d7065435961c06d3", kill_on_drop: false }`
[INFO] running `Command { std: "docker" "inspect" "92abd7de59f6720c4d28d10762c68a8a896896aa44b64c48d7065435961c06d3", kill_on_drop: false }`
[INFO] running `Command { std: "docker" "rm" "-f" "92abd7de59f6720c4d28d10762c68a8a896896aa44b64c48d7065435961c06d3", kill_on_drop: false }`
[INFO] [stdout] 92abd7de59f6720c4d28d10762c68a8a896896aa44b64c48d7065435961c06d3
[INFO] running `Command { std: "docker" "create" "-v" "/var/lib/crater-agent-workspace/builds/worker-2-tc2/target:/opt/rustwide/target:rw,Z" "-v" "/var/lib/crater-agent-workspace/builds/worker-2-tc2/source:/opt/rustwide/workdir:ro,Z" "-v" "/var/lib/crater-agent-workspace/cargo-home:/opt/rustwide/cargo-home:ro,Z" "-v" "/var/lib/crater-agent-workspace/rustup-home:/opt/rustwide/rustup-home:ro,Z" "-e" "SOURCE_DIR=/opt/rustwide/workdir" "-e" "CARGO_TARGET_DIR=/opt/rustwide/target" "-e" "CARGO_INCREMENTAL=0" "-e" "RUST_BACKTRACE=full" "-e" "RUSTFLAGS=--cap-lints=warn" "-e" "RUSTDOCFLAGS=--cap-lints=warn" "-e" "CARGO_HOME=/opt/rustwide/cargo-home" "-e" "RUSTUP_HOME=/opt/rustwide/rustup-home" "-w" "/opt/rustwide/workdir" "-m" "1610612736" "--user" "0:0" "--network" "none" "ghcr.io/rust-lang/crates-build-env/linux@sha256:86ea7c7af713d31e8cfdb68a6d0db50b5cf7cbeecde3d112f9f257f747318d36" "/opt/rustwide/cargo-home/bin/cargo" "+beta-2025-01-12" "build" "--frozen" "--message-format=json", kill_on_drop: false }`
[INFO] [stdout] c9bba1c418352159b9bee8c82bc606e9bfb2ebbcac6fd0fcc3c5b147d9b924c7
[INFO] running `Command { std: "docker" "start" "-a" "c9bba1c418352159b9bee8c82bc606e9bfb2ebbcac6fd0fcc3c5b147d9b924c7", kill_on_drop: false }`
[INFO] [stderr]    Compiling proc-macro2 v1.0.92
[INFO] [stderr]    Compiling unicode-ident v1.0.14
[INFO] [stderr]    Compiling autocfg v1.4.0
[INFO] [stderr]    Compiling libc v0.2.167
[INFO] [stderr]    Compiling cfg-if v1.0.0
[INFO] [stderr]    Compiling memchr v2.7.4
[INFO] [stderr]    Compiling thiserror v1.0.69
[INFO] [stderr]    Compiling byteorder v1.5.0
[INFO] [stderr]    Compiling serde v1.0.215
[INFO] [stderr]    Compiling version_check v0.9.5
[INFO] [stderr]    Compiling once_cell v1.20.2
[INFO] [stderr]    Compiling typenum v1.17.0
[INFO] [stderr]    Compiling ucd-trie v0.1.7
[INFO] [stderr]    Compiling rustix v0.38.41
[INFO] [stderr]    Compiling syn v1.0.109
[INFO] [stderr]    Compiling allocator-api2 v0.2.21
[INFO] [stderr]    Compiling foldhash v0.1.3
[INFO] [stderr]    Compiling dot-structures v0.1.1
[INFO] [stderr]    Compiling equivalent v1.0.1
[INFO] [stderr]    Compiling linux-raw-sys v0.4.14
[INFO] [stderr]    Compiling utf8parse v0.2.2
[INFO] [stderr]    Compiling bitflags v2.6.0
[INFO] [stderr]    Compiling dot-generator v0.2.0
[INFO] [stderr]    Compiling into-attr v0.1.1
[INFO] [stderr]    Compiling anstyle-parse v0.2.6
[INFO] [stderr]    Compiling anstyle-query v1.1.2
[INFO] [stderr]    Compiling hashbrown v0.15.2
[INFO] [stderr]    Compiling crossbeam-utils v0.8.20
[INFO] [stderr]    Compiling regex-syntax v0.8.5
[INFO] [stderr]    Compiling anstyle v1.0.10
[INFO] [stderr]    Compiling serde_json v1.0.133
[INFO] [stderr]    Compiling colorchoice v1.0.3
[INFO] [stderr]    Compiling is_terminal_polyfill v1.70.1
[INFO] [stderr]    Compiling fastrand v2.2.0
[INFO] [stderr]    Compiling im-rc v15.1.0
[INFO] [stderr]    Compiling im v15.1.0
[INFO] [stderr]    Compiling num-traits v0.2.19
[INFO] [stderr]    Compiling anstream v0.6.18
[INFO] [stderr]    Compiling clap_lex v0.7.3
[INFO] [stderr]    Compiling ryu v1.0.18
[INFO] [stderr]    Compiling aho-corasick v1.1.3
[INFO] [stderr]    Compiling heck v0.5.0
[INFO] [stderr]    Compiling strsim v0.11.1
[INFO] [stderr]    Compiling itoa v1.0.14
[INFO] [stderr]    Compiling humantime v2.1.0
[INFO] [stderr]    Compiling log v0.4.22
[INFO] [stderr]    Compiling clap_builder v4.5.21
[INFO] [stderr]    Compiling termcolor v1.4.1
[INFO] [stderr]    Compiling instant v0.1.13
[INFO] [stderr]    Compiling smallvec v1.13.2
[INFO] [stderr]    Compiling quote v1.0.37
[INFO] [stderr]    Compiling rustc-hash v1.1.0
[INFO] [stderr]    Compiling lazy_static v1.5.0
[INFO] [stderr]    Compiling syn v2.0.90
[INFO] [stderr]    Compiling bitmaps v2.1.0
[INFO] [stderr]    Compiling chrono v0.4.38
[INFO] [stderr]    Compiling symbol_table v0.4.0
[INFO] [stderr]    Compiling getrandom v0.2.15
[INFO] [stderr]    Compiling is-terminal v0.4.13
[INFO] [stderr]    Compiling rand_core v0.6.4
[INFO] [stderr]    Compiling num-integer v0.1.46
[INFO] [stderr]    Compiling num-complex v0.4.6
[INFO] [stderr]    Compiling rand_xoshiro v0.6.0
[INFO] [stderr]    Compiling num-bigint v0.4.6
[INFO] [stderr]    Compiling num-iter v0.1.45
[INFO] [stderr]    Compiling regex-automata v0.4.9
[INFO] [stderr]    Compiling sized-chunks v0.6.5
[INFO] [stderr]    Compiling egglog v0.3.0 (https://github.com/egraphs-good/egglog.git?rev=01a4a61#01a4a614)
[INFO] [stderr]    Compiling tempfile v3.14.0
[INFO] [stderr]    Compiling into-attr-derive v0.2.1
[INFO] [stderr]    Compiling num-rational v0.4.2
[INFO] [stderr]    Compiling num v0.4.3
[INFO] [stderr]    Compiling thiserror-impl v1.0.69
[INFO] [stderr]    Compiling zerocopy-derive v0.7.35
[INFO] [stderr]    Compiling serde_derive v1.0.215
[INFO] [stderr]    Compiling clap_derive v4.5.18
[INFO] [stderr]    Compiling regex v1.11.1
[INFO] [stderr]    Compiling zerocopy v0.7.35
[INFO] [stderr]    Compiling env_logger v0.10.2
[INFO] [stderr]    Compiling pest v2.7.14
[INFO] [stderr]    Compiling ppv-lite86 v0.2.20
[INFO] [stderr]    Compiling rand_chacha v0.3.1
[INFO] [stderr]    Compiling rand v0.8.5
[INFO] [stderr]    Compiling clap v4.5.21
[INFO] [stderr]    Compiling pest_meta v2.7.14
[INFO] [stderr]    Compiling pest_generator v2.7.14
[INFO] [stderr]    Compiling pest_derive v2.7.14
[INFO] [stderr]    Compiling graphviz-rust v0.6.6
[INFO] [stderr]    Compiling indexmap v2.7.0
[INFO] [stderr]    Compiling ordered-float v3.9.2
[INFO] [stderr]    Compiling egraph-serialize v0.2.0
[INFO] [stderr]    Compiling egglog-experimental v0.1.0 (/opt/rustwide/workdir)
[INFO] [stderr]     Finished `dev` profile [unoptimized + debuginfo] target(s) in 41.28s
[INFO] running `Command { std: "docker" "inspect" "c9bba1c418352159b9bee8c82bc606e9bfb2ebbcac6fd0fcc3c5b147d9b924c7", kill_on_drop: false }`
[INFO] running `Command { std: "docker" "rm" "-f" "c9bba1c418352159b9bee8c82bc606e9bfb2ebbcac6fd0fcc3c5b147d9b924c7", kill_on_drop: false }`
[INFO] [stdout] c9bba1c418352159b9bee8c82bc606e9bfb2ebbcac6fd0fcc3c5b147d9b924c7
[INFO] running `Command { std: "docker" "create" "-v" "/var/lib/crater-agent-workspace/builds/worker-2-tc2/target:/opt/rustwide/target:rw,Z" "-v" "/var/lib/crater-agent-workspace/builds/worker-2-tc2/source:/opt/rustwide/workdir:ro,Z" "-v" "/var/lib/crater-agent-workspace/cargo-home:/opt/rustwide/cargo-home:ro,Z" "-v" "/var/lib/crater-agent-workspace/rustup-home:/opt/rustwide/rustup-home:ro,Z" "-e" "SOURCE_DIR=/opt/rustwide/workdir" "-e" "CARGO_TARGET_DIR=/opt/rustwide/target" "-e" "CARGO_INCREMENTAL=0" "-e" "RUST_BACKTRACE=full" "-e" "RUSTFLAGS=--cap-lints=warn" "-e" "RUSTDOCFLAGS=--cap-lints=warn" "-e" "CARGO_HOME=/opt/rustwide/cargo-home" "-e" "RUSTUP_HOME=/opt/rustwide/rustup-home" "-w" "/opt/rustwide/workdir" "-m" "1610612736" "--user" "0:0" "--network" "none" "ghcr.io/rust-lang/crates-build-env/linux@sha256:86ea7c7af713d31e8cfdb68a6d0db50b5cf7cbeecde3d112f9f257f747318d36" "/opt/rustwide/cargo-home/bin/cargo" "+beta-2025-01-12" "test" "--frozen" "--no-run" "--message-format=json", kill_on_drop: false }`
[INFO] [stdout] e9bfc2786889b36cbe3cdb926faaa143d0c72321d634deb514b17e3fe1b801cd
[INFO] running `Command { std: "docker" "start" "-a" "e9bfc2786889b36cbe3cdb926faaa143d0c72321d634deb514b17e3fe1b801cd", kill_on_drop: false }`
[INFO] [stderr]    Compiling num-traits v0.2.19
[INFO] [stderr]    Compiling serde v1.0.215
[INFO] [stderr]    Compiling crossbeam-epoch v0.9.18
[INFO] [stderr]    Compiling either v1.13.0
[INFO] [stderr]    Compiling rayon-core v1.12.1
[INFO] [stderr]    Compiling half v2.4.1
[INFO] [stderr]    Compiling plotters-backend v0.3.7
[INFO] [stderr]    Compiling ciborium-io v0.2.2
[INFO] [stderr]    Compiling cast v0.3.0
[INFO] [stderr]    Compiling same-file v1.0.6
[INFO] [stderr]    Compiling colored v2.1.0
[INFO] [stderr]    Compiling num_cpus v1.16.0
[INFO] [stderr]    Compiling oorandom v11.1.4
[INFO] [stderr]    Compiling anes v0.1.6
[INFO] [stderr]    Compiling codspeed-criterion-compat v2.7.2
[INFO] [stderr]    Compiling glob v0.3.1
[INFO] [stderr]    Compiling walkdir v2.5.0
[INFO] [stderr]    Compiling plotters-svg v0.3.7
[INFO] [stderr]    Compiling threadpool v1.8.1
[INFO] [stderr]    Compiling ciborium-ll v0.2.2
[INFO] [stderr]    Compiling itertools v0.10.5
[INFO] [stderr]    Compiling libtest-mimic v0.6.1
[INFO] [stderr]    Compiling crossbeam-deque v0.8.5
[INFO] [stderr]    Compiling rayon v1.10.0
[INFO] [stderr]    Compiling num-integer v0.1.46
[INFO] [stderr]    Compiling num-complex v0.4.6
[INFO] [stderr]    Compiling plotters v0.3.7
[INFO] [stderr]    Compiling criterion-plot v0.5.0
[INFO] [stderr]    Compiling num-bigint v0.4.6
[INFO] [stderr]    Compiling num-iter v0.1.45
[INFO] [stderr]    Compiling num-rational v0.4.2
[INFO] [stderr]    Compiling num v0.4.3
[INFO] [stderr]    Compiling indexmap v2.7.0
[INFO] [stderr]    Compiling ordered-float v3.9.2
[INFO] [stderr]    Compiling ciborium v0.2.2
[INFO] [stderr]    Compiling serde_json v1.0.133
[INFO] [stderr]    Compiling egraph-serialize v0.2.0
[INFO] [stderr]    Compiling tinytemplate v1.2.1
[INFO] [stderr]    Compiling codspeed v2.7.2
[INFO] [stderr]    Compiling criterion v0.5.1
[INFO] [stderr]    Compiling egglog v0.3.0 (https://github.com/egraphs-good/egglog.git?rev=01a4a61#01a4a614)
[INFO] [stderr]    Compiling egglog-experimental v0.1.0 (/opt/rustwide/workdir)
[INFO] [stderr]     Finished `test` profile [unoptimized + debuginfo] target(s) in 26.64s
[INFO] running `Command { std: "docker" "inspect" "e9bfc2786889b36cbe3cdb926faaa143d0c72321d634deb514b17e3fe1b801cd", kill_on_drop: false }`
[INFO] running `Command { std: "docker" "rm" "-f" "e9bfc2786889b36cbe3cdb926faaa143d0c72321d634deb514b17e3fe1b801cd", kill_on_drop: false }`
[INFO] [stdout] e9bfc2786889b36cbe3cdb926faaa143d0c72321d634deb514b17e3fe1b801cd
[INFO] running `Command { std: "docker" "create" "-v" "/var/lib/crater-agent-workspace/builds/worker-2-tc2/target:/opt/rustwide/target:rw,Z" "-v" "/var/lib/crater-agent-workspace/builds/worker-2-tc2/source:/opt/rustwide/workdir:ro,Z" "-v" "/var/lib/crater-agent-workspace/cargo-home:/opt/rustwide/cargo-home:ro,Z" "-v" "/var/lib/crater-agent-workspace/rustup-home:/opt/rustwide/rustup-home:ro,Z" "-e" "SOURCE_DIR=/opt/rustwide/workdir" "-e" "CARGO_TARGET_DIR=/opt/rustwide/target" "-e" "CARGO_INCREMENTAL=0" "-e" "RUST_BACKTRACE=full" "-e" "RUSTFLAGS=--cap-lints=warn" "-e" "RUSTDOCFLAGS=--cap-lints=warn" "-e" "CARGO_HOME=/opt/rustwide/cargo-home" "-e" "RUSTUP_HOME=/opt/rustwide/rustup-home" "-w" "/opt/rustwide/workdir" "-m" "1610612736" "--user" "0:0" "--network" "none" "ghcr.io/rust-lang/crates-build-env/linux@sha256:86ea7c7af713d31e8cfdb68a6d0db50b5cf7cbeecde3d112f9f257f747318d36" "/opt/rustwide/cargo-home/bin/cargo" "+beta-2025-01-12" "test" "--frozen", kill_on_drop: false }`
[INFO] [stdout] 41446025a118d4ea6048e2dff5c3cea6a1edb90616ebc43ce681d621f7e745bc
[INFO] running `Command { std: "docker" "start" "-a" "41446025a118d4ea6048e2dff5c3cea6a1edb90616ebc43ce681d621f7e745bc", kill_on_drop: false }`
[INFO] [stderr]     Finished `test` profile [unoptimized + debuginfo] target(s) in 0.11s
[INFO] [stderr]      Running unittests src/lib.rs (/opt/rustwide/target/debug/deps/egglog_experimental-7833dfff5876cd9e)
[INFO] [stdout] 
[INFO] [stdout] running 0 tests
[INFO] [stdout] 
[INFO] [stdout] test result: ok. 0 passed; 0 failed; 0 ignored; 0 measured; 0 filtered out; finished in 0.00s
[INFO] [stdout] 
[INFO] [stderr]      Running unittests src/main.rs (/opt/rustwide/target/debug/deps/egglog_experimental-8947f417bbe53615)
[INFO] [stdout] 
[INFO] [stdout] running 0 tests
[INFO] [stdout] 
[INFO] [stdout] test result: ok. 0 passed; 0 failed; 0 ignored; 0 measured; 0 filtered out; finished in 0.00s
[INFO] [stdout] 
[INFO] [stderr]      Running tests/files.rs (/opt/rustwide/target/debug/deps/files-e8fd43b1bd2c9b66)
[INFO] [stdout] 
[INFO] [stdout] running 26 tests
[INFO] [stdout]   0
[INFO] [stdout]   1
[INFO] [stdout]   2
[INFO] [stdout] test for                             ... ok
[INFO] [stdout]   0
[INFO] [stdout]   1
[INFO] [stdout]   2
[INFO] [stdout]   (rational 100 1)
[INFO] [stdout] test rational                        ... ok
[INFO] [stdout] test for_resugar                     ... ok
[INFO] [stdout] test interval                        ... ok
[INFO] [stdout] test repro_unsound_htutorial         ... ok
[INFO] [stdout] test before_proofs                   ... ok
[INFO] [stdout]   (rational 100 1)
[INFO] [stdout] test before_proofs_resugar           ... ok
[INFO] [stdout] test interval_resugar                ... ok
[INFO] [stdout] test repro_unsound_htutorial_resugar ... ok
[INFO] [stdout] test rational_resugar                ... ok
[INFO] [stdout] test with_ruleset                    ... ok
[INFO] [stdout] test with_ruleset_resugar            ... ok
[INFO] [stdout]   (rational 1 1)
[INFO] [stdout]   (rational 2 1)
[INFO] [stdout]   (rational 401 201)
[INFO] [stdout]   (rational 401 201)
[INFO] [stdout]   1.9950248756218905
[INFO] [stdout]   10000.0
[INFO] [stdout]   0.0
[INFO] [stdout] test herbie_tutorial                 ... ok
[INFO] [stdout]   (rational 1 1)
[INFO] [stdout]   (rational 2 1)
[INFO] [stdout]   (rational 401 201)
[INFO] [stdout]   (rational 401 201)
[INFO] [stdout]   1.9950248756218905
[INFO] [stdout]   10000.0
[INFO] [stdout]   0.0
[INFO] [stdout] test herbie_tutorial_resugar         ... ok
[INFO] [stdout]   (Const (rational 1 1))
[INFO] [stdout]   (Const (rational 1 1))
[INFO] [stdout] test math                            ... ok
[INFO] [stdout] test herbie                          ... ok
[INFO] [stdout] test math_resugar                    ... ok
[INFO] [stdout] test herbie_resugar                  ... ok
[INFO] [stdout] test repro_unsound                   ... ok
[INFO] [stdout]   (NDArray___getitem__ (NDArray___matmul__ (NDArray___sub__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (NDArray___matmul__ (NDArray___truediv__ (astype (NDArray_vector (TupleValue___add__ (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))) (OptionalIntOrTuple_none)))) (TupleValue___add__ (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))) (OptionalIntOrTuple_none)))) (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))) (OptionalIntOrTuple_none))))))) (DType_float64)) (NDArray_scalar (Value_float (Float_rational (rational 150 1))))) (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))))) (NDArray___matmul__ (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (NDArray_scalar (Value_float (Float_rational (rational 1 147))))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (IndexKey_ndarray (NDArray___eq__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float_rational (rational 1 1))))))) (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 (NDArray_scalar (Value_float (Float_rational (rational 1 147))))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (IndexKey_ndarray (NDArray___eq__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float_rational (rational 1 1))))))) (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__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (IndexKey_ndarray (NDArray___eq__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float_rational (rational 1 1))))))) (NDArray___getitem__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (NDArray_scalar (Value_float (Float_rational (rational 1 147))))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (IndexKey_ndarray (NDArray___eq__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float_rational (rational 1 1))))))) (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 (NDArray_scalar (Value_float (Float_rational (rational 1 147))))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (IndexKey_ndarray (NDArray___eq__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float_rational (rational 1 1))))))) (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 (Int___init__ 150))) (NDArray___truediv__ (astype (NDArray_vector (TupleValue___add__ (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))) (OptionalIntOrTuple_none)))) (TupleValue___add__ (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))) (OptionalIntOrTuple_none)))) (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))) (OptionalIntOrTuple_none))))))) (DType_float64)) (NDArray_scalar (Value_float (Float_rational (rational 150 1)))))) (NDArray_scalar (Value_float (Float_rational (rational 1 2)))))) (NDArray_T (NDArray___sub__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (NDArray___matmul__ (NDArray___truediv__ (astype (NDArray_vector (TupleValue___add__ (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))) (OptionalIntOrTuple_none)))) (TupleValue___add__ (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))) (OptionalIntOrTuple_none)))) (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))) (OptionalIntOrTuple_none))))))) (DType_float64)) (NDArray_scalar (Value_float (Float_rational (rational 150 1))))) (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0))))))))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (NDArray_scalar (Value_float (Float_rational (rational 1 147))))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (IndexKey_ndarray (NDArray___eq__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float_rational (rational 1 1))))))) (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 (NDArray_scalar (Value_float (Float_rational (rational 1 147))))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (IndexKey_ndarray (NDArray___eq__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float_rational (rational 1 1))))))) (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__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (IndexKey_ndarray (NDArray___eq__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float_rational (rational 1 1))))))) (NDArray___getitem__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (NDArray_scalar (Value_float (Float_rational (rational 1 147))))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (IndexKey_ndarray (NDArray___eq__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float_rational (rational 1 1))))))) (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 (NDArray_scalar (Value_float (Float_rational (rational 1 147))))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (IndexKey_ndarray (NDArray___eq__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float_rational (rational 1 1))))))) (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 (Int___init__ 150))) (NDArray___truediv__ (astype (NDArray_vector (TupleValue___add__ (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))) (OptionalIntOrTuple_none)))) (TupleValue___add__ (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))) (OptionalIntOrTuple_none)))) (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))) (OptionalIntOrTuple_none))))))) (DType_float64)) (NDArray_scalar (Value_float (Float_rational (rational 150 1)))))) (NDArray_scalar (Value_float (Float_rational (rational 1 2)))))) (NDArray_T (NDArray___sub__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (NDArray___matmul__ (NDArray___truediv__ (astype (NDArray_vector (TupleValue___add__ (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))) (OptionalIntOrTuple_none)))) (TupleValue___add__ (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))) (OptionalIntOrTuple_none)))) (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))) (OptionalIntOrTuple_none))))))) (DType_float64)) (NDArray_scalar (Value_float (Float_rational (rational 150 1))))) (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0))))))))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (NDArray_scalar (Value_float (Float_rational (rational 1 147))))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (IndexKey_ndarray (NDArray___eq__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float_rational (rational 1 1))))))) (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 (NDArray_scalar (Value_float (Float_rational (rational 1 147))))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (IndexKey_ndarray (NDArray___eq__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float_rational (rational 1 1))))))) (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__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (IndexKey_ndarray (NDArray___eq__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float_rational (rational 1 1))))))) (NDArray___getitem__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (NDArray_scalar (Value_float (Float_rational (rational 1 147))))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (IndexKey_ndarray (NDArray___eq__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float_rational (rational 1 1))))))) (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 (NDArray_scalar (Value_float (Float_rational (rational 1 147))))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (IndexKey_ndarray (NDArray___eq__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float_rational (rational 1 1))))))) (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 (Int___init__ 150))) (NDArray___truediv__ (astype (NDArray_vector (TupleValue___add__ (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))) (OptionalIntOrTuple_none)))) (TupleValue___add__ (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))) (OptionalIntOrTuple_none)))) (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))) (OptionalIntOrTuple_none))))))) (DType_float64)) (NDArray_scalar (Value_float (Float_rational (rational 150 1)))))) (NDArray_scalar (Value_float (Float_rational (rational 1 2)))))) (NDArray_T (NDArray___sub__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (NDArray___matmul__ (NDArray___truediv__ (astype (NDArray_vector (TupleValue___add__ (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))) (OptionalIntOrTuple_none)))) (TupleValue___add__ (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))) (OptionalIntOrTuple_none)))) (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))) (OptionalIntOrTuple_none))))))) (DType_float64)) (NDArray_scalar (Value_float (Float_rational (rational 150 1))))) (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0))))))))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (NDArray_scalar (Value_float (Float_rational (rational 1 147))))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (IndexKey_ndarray (NDArray___eq__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float_rational (rational 1 1))))))) (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 (NDArray_scalar (Value_float (Float_rational (rational 1 147))))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (IndexKey_ndarray (NDArray___eq__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float_rational (rational 1 1))))))) (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__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (IndexKey_ndarray (NDArray___eq__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float_rational (rational 1 1))))))) (NDArray___getitem__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (NDArray_scalar (Value_float (Float_rational (rational 1 147))))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (IndexKey_ndarray (NDArray___eq__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float_rational (rational 1 1))))))) (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 (NDArray_scalar (Value_float (Float_rational (rational 1 147))))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (IndexKey_ndarray (NDArray___eq__ (ndarray-sqrt (NDArray___truediv__ (sum (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0)))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (square (NDArray___sub__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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))) (expand_dims (NDArray___truediv__ (sum (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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 (TupleInt___getitem__ (NDArray_shape (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) (Int___init__ 4))) (OptionalDType_some (DType_float64)) (OptionalDevice_some (NDArray_device (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 0))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 1))))))) (Int___init__ 0)))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (NDArray___truediv__ (sum (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int (Int___init__ 2))))))) (Int___init__ 0)))))) (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)))) (Int___init__ 0))))) (Int___init__ 0))))) (Int___init__ 0)))))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float_rational (rational 1 1))))))) (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)))))))
[INFO] [stdout] test stresstest_large_expr           ... ok
[INFO] [stdout] test repro_unsound_resugar           ... ok
[WARN] too much data in the log, truncating it