C4W3_Colab_T5_SQuAD_Model.ipynb

{"nbformat":4,"nbformat_minor":0,"metadata":{"accelerator":"GPU","colab":{"name":"C4W3_Colab_T5_SQuAD_Model.ipynb","provenance":[{"file_id":"1c-8KJkTySRGqCx_JjwjvXuRBTNTqEE0N","timestamp":1624413367247}],"collapsed_sections":[]},"kernelspec":{"name":"python3","display_name":"Python 3"}},"cells":[{"cell_type":"markdown","metadata":{"id":"7yuytuIllsv1"},"source":["# Assignment 3, Part 2: T5 SQuAD Model \n","\n","Welcome to the part 2 of testing the models for this week's assignment. This time we will perform decoding using the T5 SQuAD model. In this notebook we'll perform Question Answering by providing a \"Question\", its \"Context\" and see how well we get the \"Target\" answer. \n","\n","## IMPORTANT\n","\n","- As you cannot save the changes you make to this colab, you have to make a copy of this notebook in your own drive and run that. You can do so by going to `File -> Save a copy in Drive`. Close this colab and open the copy which you have made in your own drive.\n","\n","- Go to this [google drive folder](https://drive.google.com/drive/folders/1rOZsbEzcpMRVvgrRULRh1JPFpkIG_JOz?usp=sharing) named `NLP C4 W3 Data`. In the folder, next to its name use the drop down menu to select `\"Add shortcut to Drive\" -> \"My Drive\" and then press ADD SHORTCUT`. This should add a shortcut to the folder `NLP C4 W3 Data` within your own google drive. Please make sure this happens, as you'll be reading the data for this notebook from this folder.\n","\n","- Make sure your runtime is GPU (_not_ CPU or TPU). And if it is an option, make sure you are using _Python 3_. You can select these settings by going to `Runtime -> Change runtime type -> Select the above mentioned settings and then press SAVE`"]},{"cell_type":"markdown","metadata":{"id":"Db6LQW5cMSgx"},"source":["**Note: Restarting the runtime maybe required**.\n","\n","Colab will tell you if the restarting is necessary -- you can do this from the:\n","\n","Runtime > Restart Runtime\n","\n","option in the dropdown."]},{"cell_type":"markdown","metadata":{"id":"OTeAcYQo_4OW"},"source":["## Outline\n","\n","- [Part 0: Downloading and loading dependencies](#0)\n","- [Part 1: Mounting your drive for data accessibility](#1)\n","- [Part 2: Getting things ready](#2)\n","- [Part 3: Fine-tuning on SQuAD](#3)\n","    - [3.1 Loading in the data and preprocessing](#3.1)\n","    - [3.2 Decoding from a fine-tuned model](#3.2)"]},{"cell_type":"markdown","metadata":{"id":"ysxogfC1M158"},"source":["<a name='0'></a>\n","# Part 0: Downloading and loading dependencies\n","\n","Uncomment the code cell below and run it to download some dependencies that you will need. You need to download them once every time you open the colab. You can ignore the `kfac` error."]},{"cell_type":"code","metadata":{"id":"1BNZzCg0xv3R","colab":{"base_uri":"https://localhost:8080/"},"executionInfo":{"status":"ok","timestamp":1624413446493,"user_tz":-420,"elapsed":12570,"user":{"displayName":"Quý Trần Nhật","photoUrl":"","userId":"13689480600881371104"}},"outputId":"b2edd3c0-3c51-4410-e7cc-77a14f6e692a"},"source":["!pip -q install trax"],"execution_count":1,"outputs":[{"output_type":"stream","text":["\u001b[K     |████████████████████████████████| 634kB 7.4MB/s \n","\u001b[K     |████████████████████████████████| 4.3MB 37.1MB/s \n","\u001b[K     |████████████████████████████████| 153kB 55.1MB/s \n","\u001b[K     |████████████████████████████████| 256kB 39.0MB/s \n","\u001b[K     |████████████████████████████████| 4.0MB 41.7MB/s \n","\u001b[K     |████████████████████████████████| 61kB 9.0MB/s \n","\u001b[K     |████████████████████████████████| 2.5MB 45.3MB/s \n","\u001b[K     |████████████████████████████████| 1.2MB 49.2MB/s \n","\u001b[K     |████████████████████████████████| 368kB 48.5MB/s \n","\u001b[K     |████████████████████████████████| 3.3MB 41.1MB/s \n","\u001b[K     |████████████████████████████████| 901kB 41.3MB/s \n","\u001b[?25h"],"name":"stdout"}]},{"cell_type":"code","metadata":{"id":"uDhi6qLQMHzs","executionInfo":{"status":"ok","timestamp":1624413488211,"user_tz":-420,"elapsed":41721,"user":{"displayName":"Quý Trần Nhật","photoUrl":"","userId":"13689480600881371104"}}},"source":["import string\n","import t5\n","import numpy as np\n","import trax \n","from trax.supervised import decoding\n","import textwrap \n","# Will come handy later.\n","wrapper = textwrap.TextWrapper(width=70)"],"execution_count":2,"outputs":[]},{"cell_type":"markdown","metadata":{"id":"Cwr7LoXwQUW5"},"source":["<a name='1'></a>\n","# Part 1: Mounting your drive for data accessibility\n","\n","Run the code cell below and follow the instructions to mount your drive. The data is the same as used in the coursera version of the assignment."]},{"cell_type":"code","metadata":{"id":"P7ZF7KiXzQEg","colab":{"base_uri":"https://localhost:8080/"},"executionInfo":{"status":"ok","timestamp":1624413505865,"user_tz":-420,"elapsed":17656,"user":{"displayName":"Quý Trần Nhật","photoUrl":"","userId":"13689480600881371104"}},"outputId":"48557be3-e732-4fd8-b65b-fcad966f28ff"},"source":["from google.colab import drive\n","drive.mount('/content/drive/', force_remount=True)"],"execution_count":3,"outputs":[{"output_type":"stream","text":["Mounted at /content/drive/\n"],"name":"stdout"}]},{"cell_type":"markdown","metadata":{"id":"HuTyft5EBQK6"},"source":["<a name='2'></a>\n","# Part 2: Getting things ready \n","\n","Run the code cell below to ready some functions which will later help us in decoding. The code and the functions are the same as the ones you previsouly ran on the coursera version of the assignment."]},{"cell_type":"code","metadata":{"id":"LJ8s_xZ1QtkI","executionInfo":{"status":"ok","timestamp":1624413510754,"user_tz":-420,"elapsed":4892,"user":{"displayName":"Quý Trần Nhật","photoUrl":"","userId":"13689480600881371104"}}},"source":["PAD, EOS, UNK = 0, 1, 2\n","\n","def detokenize(np_array):\n","  return trax.data.detokenize(\n","      np_array,\n","      vocab_type='sentencepiece',\n","      vocab_file='sentencepiece.model',\n","      vocab_dir='/content/drive/My Drive/NLP C4 W3 Data/')\n","\n","def tokenize(s):\n","  # The trax.data.tokenize function operates on streams,\n","  # that's why we have to create 1-element stream with iter\n","  # and later retrieve the result with next.\n","  return next(trax.data.tokenize(\n","      iter([s]),\n","      vocab_type='sentencepiece',\n","      vocab_file='sentencepiece.model',\n","      vocab_dir='/content/drive/My Drive/NLP C4 W3 Data/'))\n"," \n","vocab_size = trax.data.vocab_size(\n","    vocab_type='sentencepiece',\n","    vocab_file='sentencepiece.model',\n","    vocab_dir='/content/drive/My Drive/NLP C4 W3 Data/')\n","\n","def get_sentinels(vocab_size):\n","    sentinels = {}\n","\n","    for i, char in enumerate(reversed(string.ascii_letters), 1):\n","\n","        decoded_text = detokenize([vocab_size - i]) \n","        \n","        # Sentinels, ex: <Z> - <a>\n","        sentinels[decoded_text] = f'<{char}>'\n","        \n","    return sentinels\n","\n","sentinels = get_sentinels(vocab_size)    \n","\n","\n","def pretty_decode(encoded_str_list, sentinels=sentinels):\n","    # If already a string, just do the replacements.\n","    if isinstance(encoded_str_list, (str, bytes)):\n","        for token, char in sentinels.items():\n","            encoded_str_list = encoded_str_list.replace(token, char)\n","        return encoded_str_list\n","  \n","    # We need to decode and then prettyfy it.\n","    return pretty_decode(detokenize(encoded_str_list))    "],"execution_count":4,"outputs":[]},{"cell_type":"markdown","metadata":{"id":"HEoSSKNwgDVA"},"source":["<a name='3'></a>\n","# Part 3: Fine-tuning on SQuAD\n","\n","Now let's try to fine tune on SQuAD and see what becomes of the model. For this, we need to write a function that will create and process the SQuAD `tf.data.Dataset`. Below is how T5 pre-processes SQuAD dataset as a text2text example. Before we jump in, we will have to first load in the data. \n","\n","<a name='3.1'></a>\n","### 3.1 Loading in the data and preprocessing\n","\n","You first start by loading in the dataset. The text2text example for a SQuAD example looks like:\n","\n","```\n","{\n","  'inputs': 'question: <question> context: <article>',\n","  'targets': '<answer_0>',\n","}\n","```\n","\n","The squad pre-processing function takes in the dataset and processes it using the sentencePiece vocabulary you have seen above. It generates the features from the vocab and encodes the string features. It takes on question, context, and answer, and returns \"question: Q context: C\" as input and \"A\" as target."]},{"cell_type":"code","metadata":{"id":"RcdR5Dh9UVEw","executionInfo":{"status":"ok","timestamp":1624413550538,"user_tz":-420,"elapsed":282,"user":{"displayName":"Quý Trần Nhật","photoUrl":"","userId":"13689480600881371104"}}},"source":["# Retrieve Question, C, A and return \"question: Q context: C\" as input and \"A\" as target.\n","def squad_preprocess_fn(dataset, mode='train'):\n","  return t5.data.preprocessors.squad(dataset)"],"execution_count":5,"outputs":[]},{"cell_type":"code","metadata":{"id":"0NCIejwexv4x","colab":{"base_uri":"https://localhost:8080/"},"executionInfo":{"status":"ok","timestamp":1624413652488,"user_tz":-420,"elapsed":24249,"user":{"displayName":"Quý Trần Nhật","photoUrl":"","userId":"13689480600881371104"}},"outputId":"d773d5be-0d76-45a0-d12a-c6afe04888eb"},"source":["# train generator, this takes about 1 minute\n","train_generator_fn, eval_generator_fn = trax.data.tf_inputs.data_streams(\n","  'squad/v1.1:3.0.0',\n","  data_dir='/content/drive/My Drive/NLP C4 W3 Data/data/',\n","  bare_preprocess_fn=squad_preprocess_fn,\n","  input_name='inputs',\n","  target_name='targets'\n",")\n","train_generator = train_generator_fn()\n","next(train_generator)"],"execution_count":7,"outputs":[{"output_type":"execute_result","data":{"text/plain":["(b'question: What are Guinea peppers used in ? context: Common dishes include soups and stews . Common ingredients include yams , sweet potato , cassava , onion , tomato and plantain . Spices , peppers and chilis are used in cooking , including Aframomum melegueta seeds ( Guinea pepper ) . ',\n"," b'cooking')"]},"metadata":{"tags":[]},"execution_count":7}]},{"cell_type":"code","metadata":{"id":"QGQsExH8xv40","colab":{"base_uri":"https://localhost:8080/"},"executionInfo":{"status":"ok","timestamp":1624413703470,"user_tz":-420,"elapsed":278,"user":{"displayName":"Quý Trần Nhật","photoUrl":"","userId":"13689480600881371104"}},"outputId":"79b8d97e-5346-44ac-da67-54faf02e276a"},"source":["#print example from train_generator\n","(inp, out) = next(train_generator)\n","print(inp.decode('utf8').split('context:')[0])\n","print()\n","print('context:', inp.decode('utf8').split('context:')[1])\n","print()\n","print('target:', out.decode('utf8'))"],"execution_count":8,"outputs":[{"output_type":"stream","text":["question: On what date do Jehovah Witnesses believe Jesus Christ began his rule as a king in God ' s kingdom ? \n","\n","context:  Jehovah ' s Witnesses believe that Jesus Christ began to rule in heaven as king of God ' s kingdom in October 1914 , and that Satan was subsequently ousted from heaven to the earth , resulting in \" woe \" to humanity . They believe that Jesus rules invisibly , from heaven , perceived only as a series of \" signs \" . They base this belief on a rendering of the Greek word parousia — usually translated as \" coming \" when referring to Christ — as \" presence \" . They believe Jesus ' presence includes an unknown period beginning with his inauguration as king in heaven in 1914 , and ending when he comes to bring a final judgment against humans on earth . They thus depart from the mainstream Christian belief that the \" second coming \" of Matthew 24 refers to a single moment of arrival on earth to judge humans . \n","\n","target: October 1914\n"],"name":"stdout"}]},{"cell_type":"markdown","metadata":{"id":"F_eGJHpPh-rz"},"source":["<a name='3.2'></a>\n","### 3.2 Decoding from a fine-tuned model\n","\n","You will now use an existing model that we trained for you. You will initialize, then load in your model, and then try with your own input. "]},{"cell_type":"code","metadata":{"id":"dZztX5XLa01p","executionInfo":{"status":"ok","timestamp":1624413728304,"user_tz":-420,"elapsed":289,"user":{"displayName":"Quý Trần Nhật","photoUrl":"","userId":"13689480600881371104"}}},"source":["# Initialize the model \n","model = trax.models.Transformer(\n","    d_ff = 4096,\n","    d_model = 1024,\n","    max_len = 2048,\n","    n_heads = 16,\n","    dropout = 0.1,\n","    input_vocab_size = 32000,\n","    n_encoder_layers = 24,\n","    n_decoder_layers = 24,\n","    mode='predict')  # Change to 'eval' for slow decoding."],"execution_count":9,"outputs":[]},{"cell_type":"code","metadata":{"id":"OQRAGG2fnSMe","colab":{"base_uri":"https://localhost:8080/"},"executionInfo":{"status":"ok","timestamp":1624413922729,"user_tz":-420,"elapsed":188074,"user":{"displayName":"Quý Trần Nhật","photoUrl":"","userId":"13689480600881371104"}},"outputId":"2be226c0-fd83-461b-a406-8ebc67ada206"},"source":["# load in the model\n","# this will take a minute\n","shape11 = trax.shapes.ShapeDtype((1, 1), dtype=np.int32)\n","model.init_from_file('/content/drive/My Drive/NLP C4 W3 Data/models/model_squad.pkl.gz',\n","                     weights_only=True, input_signature=(shape11, shape11))"],"execution_count":10,"outputs":[{"output_type":"execute_result","data":{"text/plain":["(((),\n","  ((), ((), ((), (), ()))),\n","  ((array([[ 1.2558754e-02,  6.0141757e-03, -9.1608437e-03, ...,\n","            -3.1235514e-04,  7.0299668e-04, -2.9498159e-03],\n","           [ 1.0156728e-02,  3.2695336e-03,  2.9464075e-03, ...,\n","             4.3419574e-04, -7.1387603e-03, -2.9692282e-03],\n","           [ 1.7245077e-03,  1.0080425e-02,  8.3300687e-04, ...,\n","             4.1082548e-03,  7.9917219e-03, -5.6605488e-03],\n","           ...,\n","           [-2.0151217e-03, -5.4900319e-04,  1.8926015e-03, ...,\n","             8.7557238e-04, -1.5533810e-03,  2.1018814e-03],\n","           [ 2.9932151e-03,  3.6891326e-03,  3.9694435e-03, ...,\n","             2.2385814e-03,  8.6587539e-04,  2.4452645e-04],\n","           [-1.4847936e-03,  1.0561392e-05,  1.8603595e-03, ...,\n","            -1.2346734e-03, -1.0969609e-03, -1.0893299e-03]], dtype=float32),\n","    (),\n","    array([[[-0.00565102,  0.00045022, -0.00682445, ..., -0.00578211,\n","             -0.00011417,  0.00068822],\n","            [-0.00333029,  0.00645752,  0.01894248, ..., -0.00234195,\n","              0.00039544,  0.00987897],\n","            [ 0.01479717, -0.00776712,  0.03264913, ...,  0.0008685 ,\n","              0.00080959,  0.00968572],\n","            ...,\n","            [ 0.00088939, -0.00497742, -0.00442023, ...,  0.00494304,\n","              0.00104518,  0.00494702],\n","            [-0.00370781, -0.00343771, -0.00449605, ...,  0.00494293,\n","              0.00104569,  0.00494692],\n","            [-0.00489605,  0.00126261, -0.00057235, ...,  0.00494282,\n","              0.00104619,  0.00494681]]], dtype=float32),\n","    (((),\n","      ((),\n","       ((array([30.099367 , 26.130331 , 25.328857 , ...,  7.812154 ,  8.624808 ,\n","                 6.8286114], dtype=float32),\n","         array([-0.00087299, -0.00058235, -0.00088101, ...,  0.00312052,\n","                -0.00080818,  0.00221783], dtype=float32)),\n","        ((),\n","         ((),\n","          ((), (((array([[-4.3157188e-04, -2.4347719e-04, -2.9364560e-04, ...,\n","                      -1.7500004e-03, -4.5013826e-04,  2.4388352e-04],\n","                     [-2.1111621e-06,  3.9882664e-04, -1.0207243e-05, ...,\n","                       2.3986120e-03, -5.3955340e-03,  2.3031587e-04],\n","                     [ 6.4703304e-06,  5.2301108e-04, -3.3134545e-04, ...,\n","                      -2.3673684e-03, -1.4795432e-03,  1.8633674e-03],\n","                     ...,\n","                     [ 1.6690802e-03, -1.1605331e-03, -2.9713646e-04, ...,\n","                      -1.9691736e-03, -6.9904758e-04, -4.0467415e-04],\n","                     [-1.0327279e-03,  8.6260418e-04,  1.8531049e-05, ...,\n","                      -1.4016510e-03, -9.5462886e-04,  1.9355286e-03],\n","                     [ 1.7429148e-03, -6.0509672e-05, -9.2528746e-05, ...,\n","                      -1.0492665e-03, -2.1385946e-03,  9.0230646e-04]], dtype=float32),\n","              array([ 0.00380308, -0.00158196,  0.00010506, ..., -0.00841766,\n","                     -0.00189627,  0.00648794], dtype=float32)),\n","             (array([[-3.1444978e-04,  8.6615153e-05, -8.2007600e-06, ...,\n","                       2.6003411e-03, -3.1285228e-03, -1.2208190e-04],\n","                     [ 2.8431948e-04,  1.6111940e-04,  2.5080860e-04, ...,\n","                       2.8331354e-03,  9.9157926e-04, -1.9291114e-03],\n","                     [ 1.5943285e-04, -1.5856593e-04,  1.7144930e-06, ...,\n","                       2.4893825e-04,  1.6703804e-04, -6.6047441e-04],\n","                     ...,\n","                     [ 3.1648963e-04, -2.4592856e-04, -4.7716277e-04, ...,\n","                      -1.1972655e-03,  1.0692390e-03,  1.1836835e-03],\n","                     [-3.2123423e-04, -9.1128248e-05, -5.6370755e-04, ...,\n","                       6.3421368e-04,  1.3061592e-03, -1.0842213e-03],\n","                     [ 3.6978460e-04, -4.2513091e-04,  3.5185556e-04, ...,\n","                       1.0738448e-03, -1.0853331e-03,  1.0564335e-03]], dtype=float32),\n","              array([-8.5014231e-05, -5.6659868e-05, -9.9033947e-05, ...,\n","                      2.7846589e-04, -1.9153926e-04, -9.3025628e-05], dtype=float32)),\n","             (array([[ 6.18701015e-05, -1.71045918e-04, -4.55081426e-05, ...,\n","                       1.52550521e-04, -9.44785570e-05, -2.12617553e-04],\n","                     [ 1.36976902e-04,  3.10328876e-04, -2.59018358e-04, ...,\n","                       3.30877170e-04, -4.02829392e-06, -2.26550212e-04],\n","                     [ 1.72594388e-04,  3.14690005e-05,  2.42589551e-04, ...,\n","                       1.49458530e-04,  1.70189989e-04, -2.95129248e-05],\n","                     ...,\n","                     [-5.65815863e-05, -1.06951491e-04, -4.67153914e-05, ...,\n","                      -3.73729214e-04, -8.28284828e-04, -4.03814829e-06],\n","                     [-3.71089322e-04,  2.05827047e-04,  3.91193225e-06, ...,\n","                      -2.59521126e-04, -2.97788240e-04, -2.16589833e-04],\n","                     [ 1.08179054e-04,  3.41356681e-05, -1.54260910e-04, ...,\n","                      -3.35288845e-04,  2.51036254e-04, -1.26958941e-04]], dtype=float32),\n","              array([-2.7946220e-04,  1.2989365e-05, -8.1811886e-04, ...,\n","                     -2.9181316e-04,  3.0646512e-05,  1.5725453e-04], dtype=float32))),\n","            (),\n","            (array([[ 6.29673887e-04,  1.48469146e-04, -4.86967474e-04, ...,\n","                      6.45221871e-06, -4.39211144e-05, -2.45969539e-04],\n","                    [-1.70880521e-04,  1.77059395e-04,  2.82179128e-04, ...,\n","                      1.10578156e-04, -1.44375692e-04,  1.24487167e-04],\n","                    [-3.28827096e-04,  2.20088405e-04,  9.50984468e-05, ...,\n","                      9.99273689e-05,  1.97962450e-04, -2.44408817e-04],\n","                    ...,\n","                    [ 6.38845260e-04,  6.78919081e-04,  1.48382569e-05, ...,\n","                      3.12072487e-04,  2.48691358e-05,  1.65735939e-04],\n","                    [ 3.71562201e-04,  6.86272921e-04,  2.04697753e-05, ...,\n","                      3.56925826e-04,  6.70575362e-04, -1.33965932e-05],\n","                    [-6.35669858e-04, -3.48464615e-04, -4.83279910e-05, ...,\n","                     -6.16587931e-04, -3.09417956e-04, -2.08492303e-04]], dtype=float32),\n","             array([-3.1733251e-04,  9.3593357e-05, -1.3749789e-04, ...,\n","                    -1.0082057e-04, -3.8561593e-05, -3.7319204e-04], dtype=float32))), ())),\n","         ()),\n","        ()))),\n","     ()),\n","    (((),\n","      ((), ((array([0.5238347, 1.1886332, 0.5743269, ..., 1.7947285, 2.5300407,\n","                2.8552978], dtype=float32),\n","         array([ 0.00199905,  0.00172037, -0.00314912, ...,  0.00115067,\n","                 0.00116991,  0.00236826], dtype=float32)),\n","        (array([[ 9.4163115e-05, -1.1634025e-05, -1.5343219e-04, ...,\n","                  9.2521186e-05,  3.2151164e-04,  3.4236923e-04],\n","                [ 1.8222468e-04,  3.0189332e-05, -1.0065578e-04, ...,\n","                  5.4638240e-05, -1.3940966e-04,  1.4581715e-04],\n","                [-6.7155030e-05,  1.5421507e-04,  3.3295856e-04, ...,\n","                 -3.9798571e-04,  3.7282234e-04,  2.0710667e-04],\n","                ...,\n","                [-3.0859569e-04,  3.8755246e-04,  1.0069945e-04, ...,\n","                 -4.4746403e-04,  3.3706633e-04,  4.7841249e-04],\n","                [ 4.9672625e-04, -1.2809524e-04,  1.2248194e-05, ...,\n","                 -8.6911133e-04, -1.2003253e-03,  8.1804588e-05],\n","                [ 3.4677191e-04,  1.5023696e-04, -1.3315928e-04, ...,\n","                  1.3406768e-04, -3.5545323e-04, -9.3046408e-05]], dtype=float32),\n","         array([ 1.21801015e-04, -2.06405948e-05,  4.65542598e-06, ...,\n","                 5.91405260e-05,  1.52499226e-04, -3.62640450e-04], dtype=float32)),\n","        ((), (), ()),\n","        (),\n","        (array([[-3.8092965e-04,  1.1512367e-04, -6.0395419e-04, ...,\n","                  1.7767822e-04,  7.2830851e-05, -2.4469206e-04],\n","                [ 5.2561139e-05, -4.3693971e-04,  3.1738239e-04, ...,\n","                 -2.4119232e-04,  8.3486237e-05,  8.4322575e-04],\n","                [-4.8965627e-05, -5.0164177e-04, -4.8636089e-04, ...,\n","                  3.7410771e-04, -1.4959754e-04,  2.5791462e-04],\n","                ...,\n","                [-3.1256845e-05,  2.4818795e-04, -2.6376647e-04, ...,\n","                 -3.7565784e-04, -2.4797730e-05, -4.0952206e-04],\n","                [ 1.7635610e-04,  2.3820589e-04,  1.5299550e-04, ...,\n","                 -4.0717373e-04,  1.1579773e-03,  6.0103962e-04],\n","                [ 2.6988881e-04, -3.2837186e-05, -6.3097285e-04, ...,\n","                  9.9588302e-05, -1.8764076e-05, -5.3928008e-05]], dtype=float32),\n","         array([-1.8629183e-04,  2.6607193e-04, -1.0361411e-04, ...,\n","                -2.3257038e-04, -1.8273492e-07, -2.4811112e-04], dtype=float32)),\n","        ()))),\n","     ()),\n","    (((),\n","      ((), ((array([32.696716, 27.04107 , 30.284327, ..., 11.741943, 12.299749,\n","                13.989752], dtype=float32),\n","         array([-0.00060839, -0.00067304,  0.00022531, ...,  0.00106999,\n","                -0.00029645,  0.00035935], dtype=float32)),\n","        ((),\n","         ((),\n","          ((),\n","           (((array([[-3.03255365e-04,  7.00464647e-04,  1.31544855e-03, ...,\n","                      -1.23043032e-03,  1.18916330e-03,  1.19882694e-04],\n","                     [-1.43565529e-03, -5.79052830e-05, -3.37050413e-04, ...,\n","                      -4.92991647e-04,  2.74257691e-05,  3.48043628e-04],\n","                     [-1.37789198e-03,  9.83980368e-04, -1.37101510e-04, ...,\n","                       5.01398521e-04, -2.11143924e-04,  3.40610481e-04],\n","                     ...,\n","                     [ 6.91906142e-04, -7.08434905e-04,  1.09908881e-03, ...,\n","                      -3.82277707e-04, -1.27544673e-03,  4.35074704e-04],\n","                     [-1.52220402e-03, -1.23144023e-03,  6.27935748e-04, ...,\n","                       1.95485089e-04,  1.70856074e-05,  9.22861291e-05],\n","                     [ 6.67870452e-04, -1.46745879e-05,  9.18632490e-04, ...,\n","                       7.30671920e-04, -2.53324077e-04,  2.73509300e-04]], dtype=float32),\n","              array([ 1.8669818e-05, -3.0020962e-04,  8.3370722e-04, ...,\n","                      8.1286032e-04, -8.5540756e-04,  3.0024940e-04], dtype=float32)),\n","             (array([[ 2.0370478e-04, -8.8629854e-04, -2.9317720e-04, ...,\n","                       5.2904023e-04, -7.7009801e-04,  2.6222080e-04],\n","                     [-4.2801685e-04, -3.3043238e-04,  6.1952876e-04, ...,\n","                      -7.9145408e-05, -4.9446506e-04, -5.9388432e-04],\n","                     [ 9.8779914e-04, -5.3726975e-04, -1.3651969e-03, ...,\n","                      -1.5675806e-06,  1.9349206e-04,  8.2170940e-04],\n","                     ...,\n","                     [-1.0716680e-03, -2.0895903e-04,  6.0847326e-04, ...,\n","                       1.0059233e-03, -6.3264341e-04, -5.8016532e-05],\n","                     [ 1.2407448e-04,  5.0007011e-04,  5.9223993e-07, ...,\n","                       9.6143296e-05, -3.7107940e-04,  4.9616589e-05],\n","                     [-1.1775552e-03, -2.0299081e-04, -6.0441979e-04, ...,\n","                       1.3821356e-05, -1.7655606e-04,  7.3406351e-05]], dtype=float32),\n","              array([-1.06751395e-05, -4.22754732e-04,  1.40146702e-04, ...,\n","                     -2.75856451e-06,  6.05368296e-05, -4.76314199e-05], dtype=float32)),\n","             (array([[ 2.52233556e-04,  2.18826171e-04, -1.80118805e-04, ...,\n","                      -7.80406699e-04,  3.17116763e-04, -5.43701521e-04],\n","                     [ 1.87278085e-04,  7.84876931e-04,  5.53296413e-04, ...,\n","                      -5.03903080e-04,  2.83014029e-04, -6.49085050e-05],\n","                     [-2.04376644e-04,  1.46949547e-04,  7.60338389e-06, ...,\n","                       1.86362086e-05, -2.47988763e-04, -4.08369815e-04],\n","                     ...,\n","                     [-1.90130770e-04,  1.11132162e-04, -1.19415527e-04, ...,\n","                       1.12639296e-04,  4.56561917e-04, -1.14643277e-04],\n","                     [ 3.64409818e-04,  3.99818702e-04, -3.96183023e-05, ...,\n","                       1.97708170e-04, -8.36716848e-04,  7.53502187e-04],\n","                     [-4.08389933e-05,  3.60762671e-04,  1.99954346e-04, ...,\n","                       8.99551960e-05,  2.30294667e-04,  4.05523489e-04]], dtype=float32),\n","              array([-1.401897e-04,  7.912255e-05, -2.540079e-04, ..., -1.472145e-04,\n","                      9.529807e-06,  4.655184e-05], dtype=float32))),\n","            (),\n","            (array([[ 1.9670793e-04,  4.0966657e-04,  5.4888841e-04, ...,\n","                     -4.9637747e-04, -1.2144730e-04,  9.4351701e-05],\n","                    [-5.2192470e-04,  1.0765801e-04, -4.4272747e-04, ...,\n","                     -3.3541393e-04, -5.1689167e-06, -2.3803292e-04],\n","                    [-2.6915231e-04,  9.4513482e-05,  5.0485716e-04, ...,\n","                     -9.6459029e-05,  3.6467638e-04,  1.8121442e-04],\n","                    ...,\n","                    [-1.0532230e-03, -1.7946365e-06,  2.6491410e-04, ...,\n","                      6.4488842e-05,  2.4364320e-04,  1.0455202e-04],\n","                    [ 5.9724855e-04, -3.6825347e-04, -7.3535228e-04, ...,\n","                      2.4639684e-04,  3.0574265e-05,  4.6086378e-04],\n","                    [ 5.9889560e-04, -1.5405774e-04, -1.8388251e-04, ...,\n","                     -3.4973072e-04,  1.0599562e-04, -1.1261975e-04]], dtype=float32),\n","             array([-1.2752577e-04,  2.1669605e-04, -4.8123038e-04, ...,\n","                     1.4240727e-04, -2.8409457e-05, -4.6137109e-04], dtype=float32))),\n","           ())),\n","         ()),\n","        ()))),\n","     ()),\n","    (((),\n","      ((),\n","       ((array([0.69710934, 1.220611  , 2.2005317 , ..., 1.2825781 , 1.0078409 ,\n","                1.48433   ], dtype=float32),\n","         array([ 3.3034150e-05,  3.0350740e-04, -4.1439576e-04, ...,\n","                 6.5384906e-05,  2.9494584e-04,  3.3946038e-04], dtype=float32)),\n","        (array([[-1.27000894e-04,  1.79621165e-05,  3.32339288e-04, ...,\n","                  3.31677205e-04, -3.76757787e-04,  1.20285666e-04],\n","                [-4.85761084e-05, -1.90941009e-05, -4.15296323e-04, ...,\n","                 -2.82616995e-04,  2.53329508e-05, -2.61942245e-04],\n","                [ 4.44692094e-04, -3.18508421e-04,  6.03853143e-04, ...,\n","                 -3.36353260e-04,  3.24784545e-04,  2.50242185e-04],\n","                ...,\n","                [ 3.91034613e-04,  3.52462666e-04,  8.86330614e-04, ...,\n","                 -6.21988089e-04, -7.26177066e-04, -1.34143862e-04],\n","                [-1.57181174e-04,  6.67092027e-05,  1.10407513e-04, ...,\n","                  6.77251955e-04,  1.17573516e-04, -1.66035039e-04],\n","                [ 2.64673523e-04,  1.48935243e-04, -1.88168793e-04, ...,\n","                  5.64743881e-04,  1.95404733e-04,  5.09674835e-04]], dtype=float32),\n","         array([-2.0326895e-04, -2.2575853e-04, -8.5278982e-05, ...,\n","                -5.2814319e-04, -1.9389650e-04, -2.4748532e-04], dtype=float32)),\n","        ((), (), ()),\n","        (),\n","        (array([[-2.22626149e-05,  6.39528967e-04, -2.78624066e-04, ...,\n","                  5.03460236e-04, -6.64068619e-04, -8.43726972e-04],\n","                [ 1.91274070e-04,  5.17344684e-04,  7.59482908e-04, ...,\n","                  3.76892800e-04,  6.06278700e-05,  9.32081530e-05],\n","                [-6.22211373e-04,  3.59953468e-04, -1.31748745e-03, ...,\n","                  1.41899148e-03,  2.74229620e-04, -6.14205434e-04],\n","                ...,\n","                [-2.53839069e-04,  2.32331629e-04,  3.10798292e-04, ...,\n","                  2.45979958e-04, -1.11239126e-04, -1.03670277e-03],\n","                [ 4.07419517e-04, -1.30704511e-03, -8.33590340e-04, ...,\n","                 -1.52671819e-05, -7.80238246e-04,  3.94844974e-04],\n","                [ 3.15775978e-04, -4.96481916e-05, -2.77606392e-04, ...,\n","                  2.65172974e-04,  2.40198555e-04,  2.45212286e-04]], dtype=float32),\n","         array([ 1.9260755e-04, -3.0339268e-04, -5.3207896e-05, ...,\n","                 1.5450713e-04, -3.6161116e-05, -2.8578506e-04], dtype=float32)),\n","        ()))),\n","     ()),\n","    (((),\n","      ((),\n","       ((array([15.224786 , 23.864214 ,  1.3948606, ..., 17.620487 , 16.569593 ,\n","                12.619397 ], dtype=float32),\n","         array([-0.0004659 , -0.00018178,  0.00433648, ...,  0.0002103 ,\n","                -0.00037691, -0.00058664], dtype=float32)),\n","        ((),\n","         ((),\n","          ((), (((array([[ 6.7734235e-04, -2.2759645e-04, -1.0357483e-04, ...,\n","                       1.6276484e-03,  1.3159042e-03, -2.2139533e-03],\n","                     [-4.8550227e-04, -1.0842655e-03, -2.3745220e-04, ...,\n","                       2.2913770e-04, -6.5989466e-04,  1.4524063e-04],\n","                     [-3.9488386e-04, -1.3688106e-04, -4.3863966e-04, ...,\n","                       2.0349685e-04, -4.6289086e-04,  3.1290425e-04],\n","                     ...,\n","                     [-1.6377638e-03, -6.8971189e-04, -1.9432166e-04, ...,\n","                      -4.3204767e-04,  1.1383863e-04,  1.2907707e-04],\n","                     [ 2.5319427e-03, -7.1031725e-05,  1.2709232e-03, ...,\n","                       1.4115656e-03, -5.8515382e-04,  4.2211745e-04],\n","                     [ 6.5897306e-04, -1.1166219e-03, -2.5955102e-04, ...,\n","                      -5.0832966e-04, -7.4549398e-04, -3.0537607e-04]], dtype=float32),\n","              array([ 0.00048413,  0.00062082,  0.00036537, ...,  0.00025241,\n","                      0.00061688, -0.00051278], dtype=float32)),\n","             (array([[-2.06367025e-04, -4.55199945e-04,  2.87924573e-04, ...,\n","                      -1.71269756e-04,  8.15413805e-05, -3.12223972e-04],\n","                     [-7.77455338e-04,  9.13600496e-04,  4.15943097e-04, ...,\n","                      -3.58262594e-04, -8.29756376e-04,  1.23954751e-03],\n","                     [-1.28687127e-03,  2.40636989e-04,  1.55876565e-03, ...,\n","                      -1.37308947e-04, -2.03843636e-04,  2.57094885e-04],\n","                     ...,\n","                     [-6.39175880e-04, -7.23856967e-04,  9.57288896e-04, ...,\n","                       1.74346834e-03,  3.91931098e-04, -7.02324789e-04],\n","                     [ 1.05381035e-03, -7.12025154e-04,  6.72194641e-04, ...,\n","                      -5.99995314e-04, -3.36863275e-04,  3.09226889e-04],\n","                     [ 4.26149578e-04,  7.70998304e-04,  1.03331535e-04, ...,\n","                      -8.26269621e-04, -5.96364669e-04,  1.60281692e-04]], dtype=float32),\n","              array([ 5.9569848e-04, -5.4753877e-05, -4.3261319e-04, ...,\n","                      2.6865533e-04, -1.6554639e-04, -6.3248708e-06], dtype=float32)),\n","             (array([[ 4.9750280e-04,  5.8603771e-05,  4.2944399e-04, ...,\n","                      -4.4910584e-04,  6.6475361e-05, -2.0611855e-05],\n","                     [-5.2373228e-04, -4.4679829e-05, -6.4430540e-05, ...,\n","                       4.6695187e-04,  1.9802310e-04,  6.5756256e-05],\n","                     [-2.1945850e-04, -2.2666043e-04, -1.8650917e-05, ...,\n","                       3.8642418e-05,  1.7398220e-04,  6.9524620e-05],\n","                     ...,\n","                     [ 6.9341906e-05, -7.0971117e-04, -4.5349318e-04, ...,\n","                       2.9805585e-04,  2.7377563e-04, -2.5568125e-04],\n","                     [ 4.0254853e-04, -1.7608650e-04, -2.5659977e-04, ...,\n","                       2.8027457e-04, -2.3807376e-04, -1.6306287e-05],\n","                     [-6.9474720e-04,  1.0212658e-03, -3.9267249e-04, ...,\n","                       2.0364215e-04, -2.4801239e-04, -2.7342213e-04]], dtype=float32),\n","              array([1.1186087e-04, 1.3407497e-04, 7.7209712e-05, ..., 4.4647993e-05,\n","                     1.9495951e-07, 1.8438903e-05], dtype=float32))),\n","            (),\n","            (array([[-1.0003134e-03, -5.3693779e-04,  5.8122969e-04, ...,\n","                     -3.5407371e-04, -3.0814972e-05,  3.3944382e-04],\n","                    [ 4.2092902e-04, -1.2424307e-04,  2.8855941e-04, ...,\n","                      5.4928096e-05, -3.5784519e-04, -2.9386798e-04],\n","                    [-4.9566303e-04, -4.8363887e-04,  7.8015422e-05, ...,\n","                     -2.1959415e-04, -3.1921835e-04,  3.1382393e-04],\n","                    ...,\n","                    [ 1.7590536e-03,  6.3103806e-07, -8.2950310e-06, ...,\n","                     -1.7224396e-04, -3.2117273e-04, -6.6700770e-04],\n","                    [ 3.2413114e-05, -2.7158263e-04, -7.5865391e-04, ...,\n","                      4.8655437e-04,  3.6567042e-04,  2.1137223e-04],\n","                    [-1.3624029e-03, -8.8185666e-04, -1.0449293e-03, ...,\n","                      1.6218008e-04, -7.8910368e-04,  4.7659775e-04]], dtype=float32),\n","             array([ 2.4177969e-05,  1.0966464e-04, -1.0302208e-03, ...,\n","                     4.4394288e-05,  2.2224344e-04, -1.8779885e-04], dtype=float32))), ())),\n","         ()),\n","        ()))),\n","     ()),\n","    (((),\n","      ((),\n","       ((array([0.78530705, 2.2454045 , 5.56903   , ..., 2.3033586 , 3.058935  ,\n","                4.2857385 ], dtype=float32),\n","         array([ 0.00075155,  0.00068171, -0.00564406, ..., -0.00019374,\n","                 0.00183371, -0.00060246], dtype=float32)),\n","        (array([[-4.5308188e-04,  4.3449894e-04, -3.5888757e-04, ...,\n","                  5.0061016e-04,  1.5662701e-04, -2.6345244e-04],\n","                [ 5.2005932e-04, -2.5961766e-04, -2.1717258e-04, ...,\n","                 -4.0076656e-04,  3.0441643e-04, -1.0022006e-03],\n","                [ 4.4196675e-04,  9.9931215e-04,  4.6434221e-04, ...,\n","                  2.4168679e-04,  6.6366536e-04, -3.1384607e-04],\n","                ...,\n","                [-2.7256328e-04,  4.7871564e-04, -8.2360796e-04, ...,\n","                  2.6555656e-04, -6.3811563e-04,  3.7797683e-04],\n","                [-7.1264069e-05, -2.4744141e-04,  7.2405406e-04, ...,\n","                 -8.2323444e-04, -3.2677312e-04,  5.0735372e-05],\n","                [ 5.5711606e-04, -2.3558861e-04,  2.9307662e-04, ...,\n","                 -1.2543975e-04,  3.0531309e-04, -4.0148714e-04]], dtype=float32),\n","         array([-5.6659733e-04, -4.5025657e-04, -3.0908632e-04, ...,\n","                -1.6657972e-04, -3.4500350e-04, -8.5724692e-05], dtype=float32)),\n","        ((), (), ()),\n","        (),\n","        (array([[ 2.4825431e-04, -2.2019882e-04,  2.1022798e-04, ...,\n","                 -5.4856710e-06, -4.4882335e-04, -3.9100234e-04],\n","                [ 1.3132388e-04, -1.2347523e-04, -1.3287504e-04, ...,\n","                  3.9670401e-04, -5.8782607e-04,  5.3684249e-05],\n","                [-8.3124323e-05,  9.3662721e-04, -2.0363969e-04, ...,\n","                 -9.1397209e-04, -5.9040805e-04, -5.5415265e-04],\n","                ...,\n","                [-3.4500513e-04, -5.6373514e-04,  1.5482469e-04, ...,\n","                  1.0224133e-03, -6.1946746e-04, -6.2585616e-04],\n","                [-5.0445134e-04, -6.5223058e-04, -4.7514384e-04, ...,\n","                  8.1744889e-05,  4.2103737e-04,  2.6452396e-04],\n","                [-1.5336181e-04,  2.4162637e-04,  6.8669440e-04, ...,\n","                 -2.6295673e-05,  2.3180660e-04, -6.1443198e-04]], dtype=float32),\n","         array([ 1.2981329e-05, -2.6870583e-04,  3.0175911e-04, ...,\n","                -4.3615528e-05,  9.1408379e-05, -9.8257660e-05], dtype=float32)),\n","        ()))),\n","     ()),\n","    (((),\n","      ((), ((array([ 8.789856, 14.37998 ,  6.419229, ..., 16.830269, 16.225073,\n","                13.759125], dtype=float32),\n","         array([-3.5606700e-04, -4.0254326e-04, -5.4005213e-06, ...,\n","                 3.9025431e-04, -2.4075794e-04, -7.6255223e-05], dtype=float32)),\n","        ((),\n","         ((),\n","          ((), (((array([[ 9.1260899e-04,  9.1099844e-04,  2.1263221e-03, ...,\n","                       3.9789619e-04,  1.8618297e-03, -1.3479226e-05],\n","                     [-1.6511462e-03, -1.0233195e-03, -9.7060495e-04, ...,\n","                      -4.6579470e-04, -1.0250111e-03,  6.1544520e-04],\n","                     [ 1.4851417e-03,  5.3111496e-03,  2.3236708e-03, ...,\n","                       6.9656002e-04, -8.1296510e-04,  4.4663440e-04],\n","                     ...,\n","                     [ 1.4614677e-03,  1.0234925e-03, -4.4898374e-04, ...,\n","                      -7.3680060e-04, -1.7911356e-03, -2.9235863e-04],\n","                     [ 1.6976208e-04, -6.7390727e-05,  7.1893079e-04, ...,\n","                       8.0825540e-04, -6.4527253e-06,  4.5715820e-04],\n","                     [-2.4489060e-04,  3.3651767e-04,  7.2240055e-04, ...,\n","                       6.4274861e-04, -7.7895922e-05, -1.2191827e-04]], dtype=float32),\n","              array([ 0.00059316, -0.00150224, -0.00062698, ..., -0.0003925 ,\n","                      0.00010891, -0.00029157], dtype=float32)),\n","             (array([[-6.8073330e-04, -4.8624075e-04,  2.7891708e-04, ...,\n","                       3.4769176e-04,  8.0000631e-05,  1.1846905e-03],\n","                     [ 6.5324228e-04,  1.7279689e-03, -1.1633735e-03, ...,\n","                       9.3499187e-04, -2.1095558e-03, -1.1933374e-04],\n","                     [-1.3604815e-03, -1.7852579e-03,  1.0558716e-03, ...,\n","                       3.9477451e-05,  3.0763436e-04,  4.7794590e-04],\n","                     ...,\n","                     [ 1.8995490e-03,  8.4738509e-04, -2.0094316e-03, ...,\n","                      -7.8744307e-04, -7.0023944e-04,  5.7707719e-05],\n","                     [-1.4252417e-03,  1.0482084e-03,  1.3880840e-03, ...,\n","                       2.4036681e-03, -1.1471408e-04,  5.8983179e-04],\n","                     [-4.3382079e-04, -1.2997079e-04,  3.4120330e-04, ...,\n","                       1.0521965e-03,  3.1574856e-04, -1.6487538e-04]], dtype=float32),\n","              array([-9.5288124e-05,  3.9799823e-05,  3.1476372e-04, ...,\n","                      1.7544067e-04,  1.8562228e-04,  1.6095446e-04], dtype=float32)),\n","             (array([[ 2.8540663e-04,  1.0185661e-04,  5.3233164e-04, ...,\n","                       5.8174209e-04, -1.4892395e-04,  3.5583776e-05],\n","                     [ 1.8003238e-04,  1.9245685e-05,  2.0575371e-04, ...,\n","                       3.8867892e-04, -2.8995163e-04,  2.8390007e-04],\n","                     [-1.6954850e-04,  9.7304564e-06, -4.0415989e-04, ...,\n","                      -4.2179271e-04,  7.0165777e-05,  1.1739363e-04],\n","                     ...,\n","                     [ 2.1000636e-04,  7.3461517e-05, -2.0076061e-04, ...,\n","                      -1.0170182e-04,  1.7579773e-04, -3.1098281e-04],\n","                     [-1.3443751e-04,  1.4308804e-04, -2.7531944e-04, ...,\n","                      -1.7062016e-04, -8.7944412e-05, -6.7822048e-06],\n","                     [ 6.9196096e-05, -1.4510939e-06,  2.1960057e-04, ...,\n","                       1.8224698e-04, -3.0010653e-05, -2.2849582e-04]], dtype=float32),\n","              array([ 1.2727926e-04,  6.0254304e-05,  1.9019248e-05, ...,\n","                      1.6713727e-04, -6.5455672e-05,  8.7141001e-05], dtype=float32))),\n","            (),\n","            (array([[-3.5209439e-04, -6.8618450e-05,  1.9146613e-04, ...,\n","                     -2.6783597e-04, -3.9455682e-04, -3.4810108e-04],\n","                    [-3.9414343e-04, -7.1680022e-04, -5.8865966e-04, ...,\n","                      2.6961439e-04, -4.5225770e-05,  2.4629696e-04],\n","                    [-4.0571994e-04, -1.5255086e-04,  6.9956260e-04, ...,\n","                     -4.2076607e-04,  2.3252693e-04, -1.5146052e-04],\n","                    ...,\n","                    [-7.4656418e-05, -3.2441062e-04, -8.8548881e-04, ...,\n","                      2.2501202e-04, -5.3419638e-04,  2.6335128e-04],\n","                    [-5.5222400e-04,  2.4152371e-04,  5.6945067e-04, ...,\n","                      9.0422080e-05, -4.5636149e-05,  3.0336608e-04],\n","                    [-5.8496592e-04,  4.8100417e-05,  4.2809962e-04, ...,\n","                      1.8676060e-04,  3.9653794e-04,  1.7984145e-04]], dtype=float32),\n","             array([ 6.9588656e-05,  4.4852452e-05, -8.1737724e-04, ...,\n","                    -9.1393544e-05, -9.9713943e-05, -1.6821742e-04], dtype=float32))), ())),\n","         ()),\n","        ()))),\n","     ()),\n","    (((),\n","      ((), ((array([2.3474765, 3.092821 , 7.765753 , ..., 3.00219  , 3.5853944,\n","                5.312737 ], dtype=float32),\n","         array([ 0.00190423, -0.00119115, -0.00619665, ...,  0.00010853,\n","                 0.00148824, -0.00110718], dtype=float32)),\n","        (array([[ 6.0477946e-04, -5.3640065e-04, -2.8533557e-06, ...,\n","                 -5.0214777e-04, -1.5646708e-04, -4.7996023e-04],\n","                [-1.1354824e-04, -1.9942349e-04,  2.0672572e-04, ...,\n","                 -3.0372670e-04, -2.0407325e-04,  2.0643222e-04],\n","                [ 1.6441452e-04,  3.2358087e-04,  9.0443791e-05, ...,\n","                 -7.1819792e-05, -2.2307477e-05,  1.4244352e-04],\n","                ...,\n","                [-8.6709741e-04,  6.1998889e-04,  4.8735866e-04, ...,\n","                 -4.4162193e-04, -4.4471704e-04,  1.8193324e-04],\n","                [ 3.3272456e-05,  1.7329134e-04, -1.3618075e-04, ...,\n","                 -7.6232059e-04, -1.2233776e-04,  1.5993217e-04],\n","                [-2.8399975e-04,  1.1660632e-03, -4.9527001e-04, ...,\n","                  5.9668277e-04, -7.1605900e-04,  3.6101267e-04]], dtype=float32),\n","         array([-0.00024256, -0.00038354, -0.00020534, ..., -0.00033277,\n","                -0.00059835, -0.00020663], dtype=float32)),\n","        ((), (), ()),\n","        (),\n","        (array([[-9.0139272e-04,  4.4048807e-04, -2.9585615e-04, ...,\n","                  1.5948529e-04,  5.6600525e-06,  4.2535985e-04],\n","                [-2.9483036e-04,  7.5791642e-04, -4.9977668e-04, ...,\n","                  5.9009006e-05, -6.4103835e-04, -7.8455091e-04],\n","                [-3.2580245e-04, -2.0196923e-04,  7.4210350e-04, ...,\n","                 -4.9337181e-05,  7.9985015e-04,  5.8308156e-04],\n","                ...,\n","                [ 1.4200064e-04,  6.1036344e-04, -5.1695440e-04, ...,\n","                 -3.1922502e-04,  1.0627274e-03, -9.1738984e-05],\n","                [-5.9441001e-05,  7.3278119e-04,  1.2231786e-03, ...,\n","                  4.7774698e-05,  1.0184589e-04,  7.1612664e-04],\n","                [-4.0704015e-04, -3.5477897e-05,  1.8371205e-04, ...,\n","                  2.4693123e-05,  9.2175929e-04,  1.3029315e-04]], dtype=float32),\n","         array([-5.8786376e-05, -3.3556856e-04,  2.4595042e-04, ...,\n","                -5.9909828e-05,  1.9455596e-05, -5.8532525e-05], dtype=float32)),\n","        ()))),\n","     ()),\n","    (((),\n","      ((),\n","       ((array([ 5.3704333,  9.328344 ,  1.8273262, ..., 10.891494 , 11.246692 ,\n","                 8.981383 ], dtype=float32),\n","         array([-5.0478854e-04,  2.5788252e-04,  1.8042872e-04, ...,\n","                 3.6075147e-04, -1.6336188e-04,  6.1726919e-06], dtype=float32)),\n","        ((),\n","         ((),\n","          ((), (((array([[-7.4086676e-04, -1.0275338e-03,  7.1184877e-05, ...,\n","                      -4.9998495e-04,  5.2904099e-04, -3.7478341e-04],\n","                     [-8.1131817e-04,  4.5278369e-04,  1.5305049e-03, ...,\n","                      -6.9500104e-04,  1.9712516e-04,  1.8850804e-03],\n","                     [ 1.2711904e-03,  8.9282636e-04, -4.3667451e-04, ...,\n","                      -3.8945489e-03, -1.8297113e-03,  1.9086552e-03],\n","                     ...,\n","                     [-3.0539627e-04, -9.9975918e-04, -2.9606794e-03, ...,\n","                      -8.6242898e-04,  3.8351866e-03, -2.1560639e-03],\n","                     [ 3.5581688e-04, -1.6732932e-04,  6.9708261e-04, ...,\n","                      -6.6523679e-04, -2.0459617e-04,  1.2299565e-03],\n","                     [ 2.3441790e-03,  1.4177827e-03, -1.6695379e-04, ...,\n","                      -7.7274570e-04,  1.1934612e-03, -1.4260965e-03]], dtype=float32),\n","              array([-0.00057101, -0.00052278, -0.00034241, ...,  0.00242893,\n","                      0.00116426, -0.00097166], dtype=float32)),\n","             (array([[-4.9932156e-04, -5.8788771e-04,  3.8653868e-04, ...,\n","                       5.7517533e-04, -5.7865412e-04, -5.8376927e-06],\n","                     [-1.2449613e-03,  6.4143044e-04,  2.2848636e-04, ...,\n","                      -3.6307116e-05,  1.6083806e-05,  2.2551210e-03],\n","                     [-1.2769419e-03, -1.9895588e-03, -1.7143427e-03, ...,\n","                       7.4528571e-04,  2.2800521e-03,  9.1841078e-04],\n","                     ...,\n","                     [-6.9745915e-04, -2.3627419e-04, -2.8073261e-04, ...,\n","                      -1.4569832e-04,  3.1700567e-03, -1.6958718e-03],\n","                     [ 2.3467415e-03,  2.3518033e-04,  1.9612135e-03, ...,\n","                      -1.1235332e-03, -1.2394176e-03,  1.7640458e-03],\n","                     [ 1.0233933e-05,  2.7448090e-04,  4.7286708e-04, ...,\n","                      -7.8773464e-04,  1.4643340e-03, -9.6175505e-04]], dtype=float32),\n","              array([-1.6215461e-04,  8.9404646e-05, -1.2093257e-04, ...,\n","                     -2.9616561e-04, -1.1133044e-05,  2.2428877e-04], dtype=float32)),\n","             (array([[-9.4536814e-04,  2.2413220e-05, -3.7012502e-04, ...,\n","                      -5.1204022e-04, -1.2080915e-05,  3.0727359e-04],\n","                     [-7.5382687e-04,  4.7919186e-04, -1.5579935e-04, ...,\n","                      -5.4410735e-05, -2.1182222e-04, -3.2882788e-04],\n","                     [-5.6147324e-05, -5.4206626e-05, -7.9547470e-05, ...,\n","                       2.6413990e-04,  2.7291776e-04, -1.3962931e-04],\n","                     ...,\n","                     [-6.6917222e-05, -2.0686128e-04,  1.3493646e-04, ...,\n","                      -1.0039699e-03,  4.8039551e-04, -1.0246890e-04],\n","                     [-1.9552138e-04, -1.1250740e-04,  2.3412662e-04, ...,\n","                      -2.1412713e-04,  8.6857079e-05,  9.8851006e-06],\n","                     [-7.3554280e-04,  1.9582946e-04, -1.5541445e-04, ...,\n","                      -9.7071825e-05,  9.2457667e-05,  4.4835958e-04]], dtype=float32),\n","              array([-6.3993939e-05, -1.1455708e-05,  4.1440773e-05, ...,\n","                     -4.4116364e-06, -3.3198226e-05, -4.0531224e-05], dtype=float32))),\n","            (),\n","            (array([[-6.38552010e-04,  4.92748048e-04, -3.94478760e-04, ...,\n","                     -5.07928606e-04,  2.21310882e-04, -2.13716965e-04],\n","                    [ 2.45619187e-04, -4.39157229e-05,  4.80546412e-04, ...,\n","                      3.35383374e-04, -6.98050426e-05, -4.93131694e-04],\n","                    [ 6.38373604e-04, -2.78811058e-04,  1.53361052e-05, ...,\n","                     -2.17589681e-04, -4.18835843e-04,  4.53960616e-04],\n","                    ...,\n","                    [ 2.87124509e-04, -2.96459970e-04, -6.30612776e-05, ...,\n","                      3.20088526e-04, -2.83053145e-04,  2.16308381e-05],\n","                    [ 7.73947220e-04, -9.90431818e-06, -2.60494853e-04, ...,\n","                      4.53263092e-05, -5.64479502e-04, -3.59138241e-04],\n","                    [ 8.86847149e-04,  1.50530206e-04, -1.05946936e-04, ...,\n","                      9.98499105e-04,  4.17486124e-04, -4.71843232e-04]], dtype=float32),\n","             array([ 1.3980660e-05,  7.5623160e-05, -4.8775555e-04, ...,\n","                     2.2045590e-05,  1.6609106e-04, -1.6488103e-04], dtype=float32))), ())),\n","         ()),\n","        ()))),\n","     ()),\n","    (((),\n","      ((), ((array([1.9799105, 2.143948 , 5.088053 , ..., 1.6863402, 2.415565 ,\n","                3.1393144], dtype=float32),\n","         array([ 0.00099579, -0.00116234, -0.00658427, ...,  0.00079678,\n","                 0.00118838, -0.00115315], dtype=float32)),\n","        (array([[-1.82302305e-04, -2.64447386e-04,  1.02497125e-03, ...,\n","                 -1.44999067e-04, -9.78182507e-05, -3.75727162e-04],\n","                [ 2.37852524e-04, -1.52729990e-04, -1.01030957e-04, ...,\n","                  1.79648487e-04, -3.27013840e-04,  1.13247101e-04],\n","                [ 3.59349564e-04,  4.33589419e-04,  4.96182511e-05, ...,\n","                  7.62161508e-04,  5.63696318e-04, -2.88656083e-05],\n","                ...,\n","                [-1.85215598e-04,  5.94802696e-05, -4.85262251e-04, ...,\n","                 -6.95769326e-04, -1.18857686e-04, -8.44666210e-04],\n","                [ 7.57851172e-04, -4.61068295e-04, -3.49803217e-04, ...,\n","                 -6.26282344e-05,  1.15131254e-04, -1.88613645e-04],\n","                [ 3.31396091e-04,  1.31617620e-04, -2.48687284e-04, ...,\n","                  6.21758227e-04, -5.11777471e-04, -1.53381596e-04]], dtype=float32),\n","         array([-0.00012331, -0.00031702, -0.00039221, ..., -0.00026611,\n","                -0.00016252, -0.00023171], dtype=float32)),\n","        ((), (), ()),\n","        (),\n","        (array([[ 9.0619382e-05, -3.8863256e-04, -5.7523028e-04, ...,\n","                  7.8232714e-04,  2.2414103e-04,  1.0527069e-03],\n","                [-2.8063901e-05, -1.1097713e-04, -1.3390744e-04, ...,\n","                 -3.9150138e-04, -2.9223677e-04, -1.7169505e-04],\n","                [-1.1893001e-03,  1.6108061e-04,  2.1547974e-05, ...,\n","                  3.2241287e-04, -8.0342455e-05,  4.1744951e-04],\n","                ...,\n","                [ 8.6912376e-05, -7.3834235e-04,  1.4142370e-06, ...,\n","                  2.6411051e-04, -1.5520213e-04, -2.4053943e-04],\n","                [-2.4146860e-04,  1.6659277e-04,  3.0538579e-04, ...,\n","                 -1.0933127e-04,  1.9036926e-04, -3.3345498e-04],\n","                [ 5.2323780e-04, -5.3697015e-04,  3.9047515e-04, ...,\n","                 -5.8397662e-05, -6.2616798e-04,  4.0589480e-04]], dtype=float32),\n","         array([-6.3876956e-05, -2.2563564e-04,  2.7758518e-04, ...,\n","                 5.2620766e-05,  3.9294224e-05, -2.2203622e-04], dtype=float32)),\n","        ()))),\n","     ()),\n","    (((),\n","      ((),\n","       ((array([ 6.0702267, 13.023816 ,  2.4872787, ..., 14.360629 , 13.795044 ,\n","                14.725142 ], dtype=float32),\n","         array([-2.8540846e-04, -3.3881977e-05,  3.1432312e-04, ...,\n","                 3.2141528e-04, -2.1852100e-04, -7.7002223e-05], dtype=float32)),\n","        ((),\n","         ((),\n","          ((), (((array([[ 3.7592748e-04, -5.9458386e-04, -3.5419158e-04, ...,\n","                      -1.4314447e-04,  3.9664621e-04, -1.4195028e-04],\n","                     [ 9.8496361e-04, -1.7457292e-04, -1.6554042e-05, ...,\n","                       7.3409181e-05,  9.7250793e-04, -2.5814536e-04],\n","                     [ 4.8006154e-04,  2.9958310e-03, -1.6592678e-03, ...,\n","                       1.9500153e-03, -1.0423320e-03,  1.5913267e-03],\n","                     ...,\n","                     [-9.7455853e-04, -3.1660878e-04,  3.0408072e-04, ...,\n","                      -1.2243934e-03, -1.0759899e-03,  2.8693854e-04],\n","                     [-8.0517275e-05, -1.3925425e-03, -6.3212158e-04, ...,\n","                       7.1168022e-04,  8.1530590e-05, -2.2543387e-05],\n","                     [-1.2587456e-04,  8.0834207e-04, -3.5770464e-04, ...,\n","                       4.9921637e-04, -1.1679878e-03,  6.4931117e-04]], dtype=float32),\n","              array([-0.00018475, -0.00094792,  0.0005907 , ..., -0.00074266,\n","                      0.00029469, -0.00062348], dtype=float32)),\n","             (array([[-4.20740980e-04,  8.80516018e-04, -3.85523948e-04, ...,\n","                       5.01447590e-04,  1.46615726e-04, -5.55390725e-04],\n","                     [-1.82764270e-04,  4.08654480e-04, -1.20325957e-03, ...,\n","                      -1.10597008e-04,  8.20307003e-04, -6.76730473e-04],\n","                     [-1.35996888e-04,  2.89385644e-05,  5.00224996e-04, ...,\n","                       1.05798164e-04,  3.90028552e-04, -8.34573875e-04],\n","                     ...,\n","                     [ 8.92034557e-04,  1.88312770e-04,  3.49512207e-04, ...,\n","                       7.19697491e-05, -4.76063782e-04,  6.98594085e-04],\n","                     [ 1.87064856e-04,  1.53641507e-04,  1.63029326e-04, ...,\n","                      -9.49601585e-04,  1.70151622e-03, -8.43970047e-04],\n","                     [-1.16235016e-04, -1.24650332e-03,  1.92903157e-04, ...,\n","                       1.07442793e-04, -7.59890361e-04,  5.48357959e-04]], dtype=float32),\n","              array([-2.7585620e-05,  1.6077589e-04, -2.8074332e-04, ...,\n","                      1.8628204e-04,  1.4013346e-04,  2.2244760e-04], dtype=float32)),\n","             (array([[-5.5929692e-04, -6.4622553e-05, -3.5481833e-04, ...,\n","                      -2.3741559e-04, -3.7899200e-04, -4.1665212e-04],\n","                     [ 1.0272168e-04, -2.7577434e-04, -2.6272549e-04, ...,\n","                       9.1537637e-05,  1.7568427e-04,  5.4703420e-04],\n","                     [-2.2890029e-04, -1.9745213e-04, -2.5367551e-04, ...,\n","                      -1.2321696e-05,  6.4184576e-05,  2.2857605e-05],\n","                     ...,\n","                     [ 3.2835646e-04, -6.6029147e-04,  1.4486618e-04, ...,\n","                       4.4886165e-05, -2.3067721e-04, -4.4754374e-06],\n","                     [-3.2886249e-04,  1.7747289e-04, -1.6373936e-04, ...,\n","                       3.3397842e-04, -1.3156926e-04,  6.5065012e-04],\n","                     [-6.7446708e-05,  1.9405689e-04,  1.8537899e-05, ...,\n","                      -4.9433194e-04,  5.6220155e-04,  8.8276975e-05]], dtype=float32),\n","              array([ 4.4441735e-05, -1.9154155e-05,  7.3337353e-05, ...,\n","                      7.8386540e-05,  8.7875387e-06,  2.5351044e-06], dtype=float32))),\n","            (),\n","            (array([[ 5.4521626e-04, -6.2044465e-04, -2.4683369e-04, ...,\n","                      1.0204878e-04,  1.8995225e-04, -3.9539696e-04],\n","                    [-2.6109000e-04,  2.1251016e-04, -1.2761774e-04, ...,\n","                      3.8428415e-04,  7.8561287e-05, -8.8559857e-05],\n","                    [-2.2839590e-04, -9.9292920e-05, -1.0911012e-05, ...,\n","                     -5.1489911e-05,  1.2901099e-04, -2.2133261e-04],\n","                    ...,\n","                    [-1.9869543e-04,  4.8661264e-04,  2.1351031e-04, ...,\n","                      4.6218056e-04, -6.6529133e-04,  6.7264866e-04],\n","                    [ 3.7999844e-04, -4.2896305e-04, -6.4845699e-05, ...,\n","                      2.9494229e-05,  6.8441601e-05, -8.9352764e-04],\n","                    [ 2.6863665e-04, -3.4086540e-04,  6.8247551e-04, ...,\n","                     -1.0877431e-04, -2.2711570e-04,  2.1057087e-04]], dtype=float32),\n","             array([-4.1676089e-05,  1.8440350e-04, -3.6362180e-04, ...,\n","                     1.3087492e-04,  1.7037305e-04, -2.1695976e-04], dtype=float32))), ())),\n","         ()),\n","        ()))),\n","     ()),\n","    (((),\n","      ((), ((array([2.62614  , 3.04635  , 7.150374 , ..., 2.7578843, 3.7894409,\n","                5.131989 ], dtype=float32),\n","         array([ 0.00092851, -0.00046922, -0.00448967, ..., -0.00022777,\n","                 0.00108639, -0.00112373], dtype=float32)),\n","        (array([[ 5.2258919e-04,  3.9280561e-04,  4.3666572e-04, ...,\n","                  3.4303369e-04, -2.0223178e-04, -1.0682585e-04],\n","                [-7.7275625e-05,  2.4577712e-05,  4.5021783e-04, ...,\n","                  1.8863097e-05, -7.3985153e-05, -7.9586584e-04],\n","                [ 1.8441980e-04,  6.1889330e-04,  3.4744071e-04, ...,\n","                  4.8474967e-04,  1.1762530e-03,  4.2134323e-04],\n","                ...,\n","                [-2.4430166e-04, -3.5872721e-04, -6.2901645e-06, ...,\n","                  2.6056223e-04, -4.5999672e-04,  8.4535830e-05],\n","                [-2.4386511e-04, -1.1941234e-04,  2.9803038e-04, ...,\n","                 -1.0115441e-04, -7.2599869e-05, -7.5948361e-04],\n","                [-4.5313782e-04, -1.2910433e-04,  3.8054120e-04, ...,\n","                 -3.0052266e-04, -3.2460428e-04,  1.7408124e-04]], dtype=float32),\n","         array([-0.00021205, -0.00039164, -0.00018607, ..., -0.00034302,\n","                -0.0007997 , -0.00029986], dtype=float32)),\n","        ((), (), ()),\n","        (),\n","        (array([[-3.39678081e-04,  9.78523094e-05, -2.07506775e-04, ...,\n","                  6.64983352e-04,  5.81375498e-04, -1.10201574e-04],\n","                [ 2.75517777e-05,  8.24528572e-04, -3.45612440e-04, ...,\n","                  4.14488350e-05,  7.37924245e-04,  2.42266629e-04],\n","                [ 2.14549393e-04,  3.26457346e-04, -5.48058524e-05, ...,\n","                 -2.68940221e-05,  1.59566189e-04,  6.02038519e-04],\n","                ...,\n","                [ 1.43695521e-04,  2.85862945e-04, -5.14818705e-04, ...,\n","                 -4.75300156e-04, -9.96624003e-05,  6.15419238e-04],\n","                [-2.07773890e-04, -4.05247083e-05,  9.06618836e-04, ...,\n","                  1.83087643e-04,  8.49061394e-07,  6.06932328e-04],\n","                [-1.46645389e-05,  8.53511738e-04,  2.51541118e-04, ...,\n","                 -2.47173266e-05,  9.53371404e-04, -1.02356577e-03]], dtype=float32),\n","         array([ 1.6190026e-05, -2.7455218e-04,  1.7339476e-04, ...,\n","                -3.6200789e-05,  9.5504860e-05, -7.7588498e-05], dtype=float32)),\n","        ()))),\n","     ()),\n","    (((),\n","      ((), ((array([ 0.92440236,  0.5778936 , -0.00348109, ...,  0.36729974,\n","                 0.11748073,  0.16800368], dtype=float32),\n","         array([-1.2297453e-03,  1.8152746e-04,  1.7407407e-04, ...,\n","                 2.3522638e-04, -1.6628470e-05,  3.1645354e-04], dtype=float32)),\n","        ((),\n","         ((),\n","          ((),\n","           (((array([[-8.35390005e-04, -6.15628029e-04, -4.98490175e-04, ...,\n","                      -8.41980625e-04,  1.00392255e-03, -9.91978915e-04],\n","                     [ 9.78935394e-04,  1.10485812e-03,  8.08600686e-04, ...,\n","                       9.51523427e-04, -1.24554231e-03,  1.17461907e-03],\n","                     [-4.55148169e-04, -4.32351226e-04,  3.34638113e-04, ...,\n","                       8.32880789e-04,  3.63816624e-04,  2.23505704e-04],\n","                     ...,\n","                     [ 6.33849704e-04,  7.81339884e-04, -1.19816716e-04, ...,\n","                      -3.32849915e-04, -4.84595483e-04,  1.19944547e-04],\n","                     [-3.08659859e-04, -4.38775023e-04,  5.82786961e-05, ...,\n","                      -1.50839041e-04, -4.84382006e-04,  3.16630030e-04],\n","                     [ 4.83994336e-06, -1.31895789e-03, -8.95677891e-04, ...,\n","                       3.43177417e-05,  4.45864978e-04,  1.86135308e-04]], dtype=float32),\n","              array([ 0.02199816,  0.00582759, -0.00159745, ..., -0.00962365,\n","                     -0.00333868, -0.00420395], dtype=float32)),\n","             (array([[-0.00307207, -0.00143617, -0.00046601, ...,  0.00301107,\n","                       0.00281884,  0.00046478],\n","                     [ 0.00149133,  0.00028779,  0.00021026, ..., -0.00063552,\n","                      -0.00190586,  0.00069465],\n","                     [ 0.00032332, -0.00107059, -0.0010887 , ..., -0.00140616,\n","                       0.00084278, -0.00090588],\n","                     ...,\n","                     [ 0.001112  , -0.00292512, -0.00268941, ..., -0.00020068,\n","                      -0.00091587,  0.00033156],\n","                     [-0.00332366, -0.00094162, -0.00059136, ..., -0.00062563,\n","                      -0.00082546,  0.00041132],\n","                     [ 0.0021972 ,  0.00221196,  0.00054254, ..., -0.00111023,\n","                      -0.00092206,  0.00064278]], dtype=float32),\n","              array([ 8.8620698e-05, -2.8446020e-04, -4.7931520e-05, ...,\n","                     -4.5112982e-05,  1.3019578e-04, -1.9251998e-04], dtype=float32)),\n","             (array([[ 1.10698835e-04, -2.35111860e-04,  3.00438096e-05, ...,\n","                      -2.46975978e-04, -5.32121157e-06, -5.66776260e-04],\n","                     [-3.32706986e-04, -2.63656384e-05, -1.71463704e-04, ...,\n","                      -6.60457416e-04, -1.60847150e-04, -2.34089748e-04],\n","                     [ 3.47308407e-04,  1.72081462e-04,  2.82052672e-04, ...,\n","                       3.16145306e-05,  2.70759301e-05,  5.77008686e-05],\n","                     ...,\n","                     [ 2.70454504e-04,  7.38555507e-04,  8.88686827e-06, ...,\n","                       7.65412915e-05,  1.23030652e-04, -6.79727891e-05],\n","                     [-1.47790735e-04, -3.04605375e-04, -3.62982712e-04, ...,\n","                      -1.71485401e-04, -2.61876376e-05, -6.90356246e-04],\n","                     [ 3.33653588e-05,  1.01657053e-04, -4.96661341e-05, ...,\n","                      -1.06247928e-04, -1.46518514e-05,  2.11110309e-04]], dtype=float32),\n","              array([ 8.6430467e-05, -5.5680026e-05, -6.2208506e-05, ...,\n","                     -4.7528101e-05,  6.9902280e-06, -9.7724376e-05], dtype=float32))),\n","            (),\n","            (array([[-3.4405448e-04,  4.4724761e-04, -2.2101757e-04, ...,\n","                     -1.6718949e-04, -9.5500855e-07,  1.5353019e-04],\n","                    [ 4.9189769e-04,  2.6888310e-04,  4.5197344e-04, ...,\n","                     -2.9054450e-04,  9.2572365e-05,  3.0869947e-04],\n","                    [-2.0533941e-04,  2.4213131e-04, -8.6473668e-04, ...,\n","                      2.3036757e-05,  5.7989655e-05,  5.2820600e-04],\n","                    ...,\n","                    [ 3.1109300e-04,  1.0878706e-03,  6.4665882e-04, ...,\n","                     -2.7550838e-04,  2.8531812e-04, -1.3149690e-04],\n","                    [-1.9046682e-04,  5.6655670e-05, -2.1231682e-04, ...,\n","                     -7.5068419e-05, -3.5430127e-04,  1.2795307e-04],\n","                    [ 9.9062570e-04,  3.8681400e-04,  2.5549057e-04, ...,\n","                      2.3219813e-04,  5.8237778e-04, -6.1428972e-04]], dtype=float32),\n","             array([-3.82253238e-05,  9.11099341e-05, -2.90908705e-04, ...,\n","                    -1.15323106e-04,  1.20565128e-04, -1.49749132e-04], dtype=float32))),\n","           ())),\n","         ()),\n","        ()))),\n","     ()),\n","    (((),\n","      ((), ((array([2.3767822, 5.0554156, 9.561314 , ..., 4.2453637, 5.8289104,\n","                7.715898 ], dtype=float32),\n","         array([ 0.00014967, -0.0001321 , -0.00293456, ..., -0.00032235,\n","                 0.00070261, -0.00056657], dtype=float32)),\n","        (array([[ 2.71306053e-04,  2.79440545e-04, -7.64942524e-05, ...,\n","                  1.12099086e-04, -4.65076337e-05,  2.28520585e-04],\n","                [-3.61270068e-04,  3.39609687e-04, -5.70566335e-04, ...,\n","                  2.81483779e-04,  4.20446449e-05,  4.01460100e-04],\n","                [ 1.60630698e-05,  3.58690304e-04,  3.59135185e-04, ...,\n","                  3.25469417e-04,  6.62359933e-04,  2.36301363e-04],\n","                ...,\n","                [-4.65389458e-04,  2.74760067e-04, -4.34320886e-04, ...,\n","                  1.46009013e-04, -8.83754838e-05,  2.46441952e-04],\n","                [ 4.16166964e-04,  8.18872722e-05, -6.51304726e-04, ...,\n","                  2.92225654e-04, -5.03925374e-04, -7.29900145e-04],\n","                [-3.68613473e-05,  4.44489269e-05,  3.12883523e-04, ...,\n","                 -1.43895531e-05,  2.28802994e-04,  2.21358947e-04]], dtype=float32),\n","         array([-0.00018139, -0.00034379, -0.00030583, ..., -0.00042141,\n","                -0.00041596, -0.00035706], dtype=float32)),\n","        ((), (), ()),\n","        (),\n","        (array([[-8.9771886e-05, -5.6689518e-04, -4.0436030e-06, ...,\n","                 -2.4772944e-05, -4.6765473e-04, -1.1623533e-04],\n","                [ 8.6653017e-05, -3.1077300e-04,  7.2070817e-04, ...,\n","                  3.4190444e-04, -5.1557994e-04, -2.3712176e-04],\n","                [-1.0804240e-05,  4.8371946e-04,  1.2399631e-05, ...,\n","                 -7.3556654e-04, -9.7294431e-04, -1.8459781e-05],\n","                ...,\n","                [ 4.3946508e-04, -3.1385920e-04,  2.6509337e-05, ...,\n","                  5.0586555e-04, -2.1399853e-04, -5.1744457e-04],\n","                [-1.7034503e-05,  4.8554892e-04, -4.2255930e-04, ...,\n","                  4.2982858e-05, -1.3488534e-04,  4.1114195e-04],\n","                [ 2.6223389e-04,  3.9522292e-04,  4.3930300e-04, ...,\n","                 -6.7366409e-04,  1.4956231e-04, -2.2704933e-04]], dtype=float32),\n","         array([-1.22051184e-04, -1.70506610e-04,  2.61635636e-04, ...,\n","                -3.99750315e-05, -1.48427644e-05,  9.70249021e-05], dtype=float32)),\n","        ()))),\n","     ()),\n","    (((),\n","      ((),\n","       ((array([4.9663787 , 3.8472435 , 0.02258645, ..., 3.7356253 , 3.6332204 ,\n","                4.6976933 ], dtype=float32),\n","         array([-2.9707895e-04,  6.9403060e-05,  3.5097561e-04, ...,\n","                 2.9931703e-04, -2.5862770e-04,  1.3634717e-04], dtype=float32)),\n","        ((),\n","         ((),\n","          ((), (((array([[-4.4454439e-04,  8.9559978e-04, -4.3184453e-04, ...,\n","                      -3.7786504e-05, -1.4994039e-04,  1.2962674e-03],\n","                     [-3.7504919e-04, -8.8608410e-04,  9.1900304e-04, ...,\n","                      -1.3892360e-04,  2.1451419e-04, -8.4102263e-05],\n","                     [ 2.4241311e-04, -5.5982373e-06,  1.4017572e-03, ...,\n","                      -9.4674341e-04, -1.1409468e-03,  2.7284300e-04],\n","                     ...,\n","                     [-1.9468264e-04,  2.0382130e-03, -2.3326185e-04, ...,\n","                       1.5302422e-03, -3.2953622e-03,  3.4290689e-04],\n","                     [-1.5804805e-03,  1.2345061e-03,  4.7326481e-04, ...,\n","                      -1.4464166e-03,  1.0033821e-03, -1.4933517e-03],\n","                     [ 9.3406555e-04, -1.0777826e-03, -8.9850608e-04, ...,\n","                       7.7128730e-04, -8.1012980e-04, -1.6290514e-04]], dtype=float32),\n","              array([-0.00539447,  0.00397303, -0.00310854, ..., -0.00109371,\n","                     -0.00102362, -0.00330223], dtype=float32)),\n","             (array([[-8.2311052e-04,  3.9343748e-04,  4.6394163e-04, ...,\n","                       2.2781718e-05,  2.1390945e-03,  2.4587728e-04],\n","                     [-6.3313596e-04,  5.0687402e-05,  1.1401094e-03, ...,\n","                       1.7956356e-03, -2.8363260e-04, -2.8983341e-04],\n","                     [ 5.0502358e-04,  6.1693648e-04, -5.9424120e-04, ...,\n","                      -1.3819947e-03,  9.3963303e-05, -7.7621854e-04],\n","                     ...,\n","                     [-4.6724617e-04,  7.7842548e-04, -7.6543965e-04, ...,\n","                       1.6738344e-03, -3.0906219e-03,  9.4209780e-04],\n","                     [ 2.4022353e-03,  2.6014980e-04, -1.8730602e-03, ...,\n","                      -7.3569216e-04,  1.9473607e-03, -8.4827520e-04],\n","                     [ 6.7529955e-04,  1.5482472e-04,  3.3582738e-04, ...,\n","                       4.6755347e-04, -9.0981502e-04, -7.6820963e-04]], dtype=float32),\n","              array([-5.3242708e-05, -3.0864176e-04,  1.3772116e-04, ...,\n","                     -1.7808996e-04,  5.9571223e-05, -1.5370686e-04], dtype=float32)),\n","             (array([[ 8.9783745e-05, -9.6214389e-06, -3.5635263e-04, ...,\n","                       4.2642318e-04, -4.1928975e-05, -3.4094913e-04],\n","                     [-6.1126229e-05,  7.9676090e-04, -8.8198384e-04, ...,\n","                       2.5310169e-04,  8.2204910e-04,  1.1944347e-03],\n","                     [ 7.2228540e-05,  8.2265060e-06, -7.9873047e-05, ...,\n","                       2.1484123e-04,  4.6427434e-05, -1.9663692e-04],\n","                     ...,\n","                     [ 2.5762588e-04, -1.3639923e-04, -4.2132230e-04, ...,\n","                      -1.8322669e-04,  2.6951355e-04, -7.5959571e-05],\n","                     [ 4.9173029e-04,  2.5092001e-04, -2.1997049e-04, ...,\n","                       2.4915644e-04,  5.3095230e-04, -2.2859754e-04],\n","                     [-4.5111118e-04,  2.6173308e-04,  2.6624373e-04, ...,\n","                       5.0337357e-04, -1.0100696e-03, -7.2943905e-05]], dtype=float32),\n","              array([-1.7169868e-05,  5.3237516e-05, -2.0964369e-05, ...,\n","                      8.0857324e-05,  1.6590040e-05,  8.9315639e-05], dtype=float32))),\n","            (),\n","            (array([[-2.3951921e-04, -5.0974113e-04, -9.0085814e-05, ...,\n","                      5.1426771e-04, -2.5211726e-04, -3.5553222e-04],\n","                    [ 1.0717501e-04, -1.0329623e-03,  8.0419995e-04, ...,\n","                      3.0611680e-04,  2.5633560e-04, -2.7058178e-04],\n","                    [ 1.5383255e-04,  1.2920818e-03, -1.2977271e-04, ...,\n","                     -3.3396389e-04, -1.9622059e-05,  4.3955815e-04],\n","                    ...,\n","                    [ 1.6872371e-04, -1.1560149e-03,  1.7085961e-04, ...,\n","                     -4.6873280e-05, -1.6170196e-04,  1.3278669e-05],\n","                    [-3.3303929e-04,  3.1703323e-05,  6.9866463e-04, ...,\n","                      6.8076461e-04, -8.9798246e-05,  4.2289501e-04],\n","                    [ 2.3002837e-04, -4.4338434e-05,  5.8764778e-04, ...,\n","                      1.9121332e-04,  4.8179415e-04,  3.6247089e-05]], dtype=float32),\n","             array([ 1.0143175e-04, -9.9614699e-05, -1.9468633e-04, ...,\n","                    -7.0662638e-05,  8.6554202e-05, -1.3782816e-04], dtype=float32))), ())),\n","         ()),\n","        ()))),\n","     ()),\n","    (((),\n","      ((), ((array([2.4889114, 5.669472 , 9.870793 , ..., 4.2971544, 5.8893085,\n","                9.398775 ], dtype=float32),\n","         array([ 2.1439244e-04, -3.8222701e-05, -2.2454343e-03, ...,\n","                -5.7109841e-04,  8.4914500e-04, -8.3700195e-04], dtype=float32)),\n","        (array([[ 1.55469374e-04, -6.02721702e-04, -1.37996030e-04, ...,\n","                  5.96607802e-04,  4.48662497e-04,  3.57873301e-04],\n","                [ 7.47840168e-05,  3.75272473e-04, -4.12051770e-04, ...,\n","                  3.36537079e-04,  6.05331967e-04,  4.86152749e-06],\n","                [ 3.22891749e-04,  2.20042086e-04,  4.28095198e-04, ...,\n","                  3.26137728e-04,  3.72713432e-04,  4.22815268e-04],\n","                ...,\n","                [ 1.21071644e-04, -9.29442249e-05,  6.85698236e-04, ...,\n","                 -1.24945393e-04,  6.19566825e-04, -1.27496256e-04],\n","                [-1.95254193e-04,  4.77042224e-04, -1.92086329e-04, ...,\n","                  1.67802456e-04, -3.78012541e-04, -3.46457746e-05],\n","                [-8.44037932e-05,  6.94474322e-04, -1.01004240e-04, ...,\n","                 -4.88898833e-04, -2.60827102e-04,  3.77972639e-04]], dtype=float32),\n","         array([-3.7074551e-05, -1.5058093e-04, -2.3765404e-04, ...,\n","                -1.5497750e-04, -1.9667775e-04, -1.5842066e-04], dtype=float32)),\n","        ((), (), ()),\n","        (),\n","        (array([[-1.96736830e-04,  1.11966045e-04, -1.97843823e-04, ...,\n","                  4.06381296e-05,  6.62950333e-05, -3.61179445e-05],\n","                [ 6.98328367e-05,  8.86805210e-05, -4.15074901e-04, ...,\n","                  1.67851656e-04, -1.53111741e-05,  2.42167996e-04],\n","                [-1.14110393e-04, -9.30202441e-05,  4.22829064e-04, ...,\n","                 -7.10021297e-04,  2.88811367e-04,  3.03030945e-04],\n","                ...,\n","                [-2.02130148e-04, -2.78049149e-04,  1.84635239e-04, ...,\n","                  1.95960110e-05,  7.79467984e-04,  3.42879357e-04],\n","                [ 5.03330084e-04,  2.16849527e-04, -3.95608658e-04, ...,\n","                 -1.79896597e-04,  2.33969462e-04,  4.46172169e-04],\n","                [-5.44116716e-04, -2.81108165e-04,  4.43510507e-05, ...,\n","                 -3.77168995e-04, -1.75583220e-04,  6.10358955e-04]], dtype=float32),\n","         array([2.6390979e-05, 5.6089419e-05, 3.5286543e-04, ..., 5.4912209e-05,\n","                6.6018489e-05, 1.4151586e-04], dtype=float32)),\n","        ()))),\n","     ()),\n","    (((),\n","      ((),\n","       ((array([1.4925183 , 1.4451425 , 0.00487194, ..., 1.4232125 , 1.1225982 ,\n","                0.8456772 ], dtype=float32),\n","         array([-8.0305286e-04, -1.4946537e-05, -7.1497205e-05, ...,\n","                 6.8149639e-05, -3.4005436e-05,  3.6217598e-04], dtype=float32)),\n","        ((),\n","         ((),\n","          ((), (((array([[-9.3839777e-04,  9.5154677e-04, -9.3217595e-06, ...,\n","                       1.6394914e-03,  2.4852747e-04, -5.9493404e-04],\n","                     [ 1.9258392e-03, -1.8800541e-03, -1.7326014e-03, ...,\n","                       1.6444002e-04, -1.5326118e-03,  3.4245639e-04],\n","                     [-5.0447270e-04,  6.6478038e-04,  3.5883870e-04, ...,\n","                      -5.2752270e-04, -3.6134847e-04,  6.6663278e-04],\n","                     ...,\n","                     [-1.6770450e-03,  1.4862457e-03,  1.0529659e-03, ...,\n","                      -9.8893733e-04,  1.0733355e-05,  8.4710389e-04],\n","                     [ 1.9010069e-03, -1.7935829e-03, -1.4792623e-03, ...,\n","                      -9.3188754e-04, -9.0590358e-04,  7.2131946e-04],\n","                     [-7.2783779e-04,  9.5738104e-04,  1.1445195e-03, ...,\n","                      -1.3841855e-03, -2.7871292e-04,  1.4465511e-03]], dtype=float32),\n","              array([ 0.01064753, -0.00937543, -0.00845589, ..., -0.03108204,\n","                      0.00816802,  0.0118936 ], dtype=float32)),\n","             (array([[-0.00518132,  0.00574549,  0.00543753, ...,  0.00284494,\n","                       0.00063802, -0.00107186],\n","                     [ 0.00133775, -0.00175349, -0.00174377, ..., -0.00184928,\n","                      -0.00235165,  0.00201606],\n","                     [ 0.00057864, -0.00082236, -0.0002683 , ..., -0.00088597,\n","                      -0.00074153,  0.00111339],\n","                     ...,\n","                     [ 0.00283221, -0.00327891, -0.00277599, ..., -0.00246544,\n","                      -0.00039546,  0.00188054],\n","                     [ 0.00173867, -0.00123592, -0.00232906, ...,  0.00064395,\n","                       0.00015047, -0.00030782],\n","                     [ 0.00347637, -0.00339703, -0.00267399, ..., -0.00229358,\n","                       0.00049528, -0.00011855]], dtype=float32),\n","              array([ 3.6433801e-05, -8.4124469e-05, -6.2562183e-05, ...,\n","                     -5.6298555e-05, -4.8238033e-04,  1.2062683e-04], dtype=float32)),\n","             (array([[-3.64163716e-04,  1.01461075e-03, -2.42441616e-04, ...,\n","                       2.94947677e-04, -1.52495690e-04,  4.25237668e-04],\n","                     [ 4.21794044e-04, -1.25108505e-04,  6.99515338e-04, ...,\n","                      -1.57063012e-04,  7.49927131e-05,  1.17053780e-04],\n","                     [ 1.17810596e-04, -7.10559543e-05, -4.64463337e-05, ...,\n","                      -2.30960010e-04,  1.49874668e-05, -4.75331355e-04],\n","                     ...,\n","                     [ 1.65956465e-06, -5.56763669e-04,  9.93079011e-05, ...,\n","                       2.30068727e-05, -1.63238976e-04,  2.90234835e-04],\n","                     [-2.94348487e-04, -5.75817365e-04,  8.50926663e-05, ...,\n","                      -1.85681623e-04, -7.27968145e-05,  5.05156640e-04],\n","                     [ 5.04208787e-04,  2.93071826e-05, -1.08078631e-04, ...,\n","                      -1.49499001e-05, -1.78882052e-04, -3.90535075e-04]], dtype=float32),\n","              array([-5.0734787e-05,  1.6358799e-05, -8.2440183e-06, ...,\n","                     -3.0235968e-05,  2.8174696e-05,  9.7673357e-05], dtype=float32))),\n","            (),\n","            (array([[-1.3258417e-04, -6.9847086e-04, -7.6428107e-05, ...,\n","                     -1.1698998e-04,  2.5503256e-04, -1.1973612e-04],\n","                    [-7.7512123e-06,  5.9061527e-04,  3.3906623e-04, ...,\n","                      5.5094686e-04,  2.9856921e-04,  3.4747287e-04],\n","                    [-3.3372417e-04, -8.5657358e-04, -5.9055566e-04, ...,\n","                     -1.4871226e-04,  2.6759869e-04, -2.0524960e-04],\n","                    ...,\n","                    [-4.5041859e-04, -2.0288385e-04,  1.9347272e-04, ...,\n","                     -3.2718675e-04, -7.0991293e-05, -1.5524351e-04],\n","                    [ 1.2895786e-04,  3.2688054e-05,  1.8538687e-04, ...,\n","                      5.1472004e-04,  4.5120516e-05, -2.3008286e-05],\n","                    [-1.8960478e-04,  5.8432648e-05,  2.2565461e-04, ...,\n","                     -4.3270006e-04, -6.1529590e-04,  6.7728740e-04]], dtype=float32),\n","             array([ 2.2828108e-05, -1.6006501e-04,  8.8431290e-05, ...,\n","                    -7.3556571e-06,  1.3376076e-05, -6.1449959e-05], dtype=float32))), ())),\n","         ()),\n","        ()))),\n","     ()),\n","    (((),\n","      ((),\n","       ((array([ 2.2846518,  6.777305 , 10.55696  , ...,  3.3969877,  5.443893 ,\n","                11.4346485], dtype=float32),\n","         array([-0.000303  ,  0.00022268, -0.00193278, ..., -0.0001573 ,\n","                 0.00030572, -0.00086539], dtype=float32)),\n","        (array([[ 1.9761898e-04,  5.4563774e-04, -2.6804613e-04, ...,\n","                  1.4375841e-04, -5.0346222e-05,  3.5105253e-04],\n","                [ 7.4449008e-07,  2.9727910e-04,  2.2557241e-04, ...,\n","                  3.0574600e-05, -1.4646644e-04,  6.6803372e-04],\n","                [ 4.0280336e-04, -1.6203565e-05,  3.0327073e-04, ...,\n","                  3.2403524e-04,  5.0076476e-04, -6.3568291e-06],\n","                ...,\n","                [ 1.9813304e-04,  2.3901268e-04,  1.9655803e-04, ...,\n","                 -2.2811821e-04, -6.4012391e-05,  5.9476151e-04],\n","                [ 4.2484610e-04,  1.9637732e-04, -1.4419077e-04, ...,\n","                  5.1956315e-04, -8.9582070e-05, -5.6501325e-05],\n","                [ 1.5575842e-04,  3.6789078e-04, -2.6203421e-04, ...,\n","                  1.4232012e-04, -1.5839197e-04, -2.4502218e-05]], dtype=float32),\n","         array([-3.4109016e-05, -1.4646491e-04, -1.9593719e-04, ...,\n","                -1.4810359e-04, -2.5383240e-04, -2.4372803e-04], dtype=float32)),\n","        ((), (), ()),\n","        (),\n","        (array([[ 2.3709983e-04,  3.0128323e-04, -2.0704568e-04, ...,\n","                  2.1108914e-04, -2.6371589e-04,  1.6335795e-04],\n","                [ 7.1131712e-05,  1.7193466e-04, -1.6442715e-04, ...,\n","                  9.3500850e-05,  9.0878484e-06, -1.6806574e-04],\n","                [ 6.9756521e-04, -3.9850656e-04,  4.9270218e-04, ...,\n","                 -2.0548901e-04, -5.5055757e-04,  2.2925038e-04],\n","                ...,\n","                [ 4.7211509e-04, -3.0429801e-04, -2.6470242e-04, ...,\n","                 -4.8616331e-04,  4.1114251e-04, -3.4436077e-04],\n","                [-2.0988226e-04,  5.9987482e-04,  3.4659609e-04, ...,\n","                  2.3395154e-05,  8.6397013e-05,  3.5611482e-04],\n","                [-9.5856907e-05, -1.1604069e-04,  2.7214239e-05, ...,\n","                 -2.9638159e-04, -2.9222132e-04,  1.5324421e-04]], dtype=float32),\n","         array([-4.27760169e-05,  5.53439131e-05,  3.45270470e-04, ...,\n","                -5.13388586e-05,  1.17925694e-04,  1.44725869e-04], dtype=float32)),\n","        ()))),\n","     ()),\n","    (((),\n","      ((),\n","       ((array([11.035136 , 13.986338 ,  4.096779 , ..., 11.3088665, 15.180344 ,\n","                14.329832 ], dtype=float32),\n","         array([ 6.9466645e-05, -3.0918138e-05,  2.5264430e-04, ...,\n","                 2.9235083e-04,  2.4956203e-04,  1.2811532e-04], dtype=float32)),\n","        ((),\n","         ((),\n","          ((), (((array([[ 2.0492859e-03,  1.0464811e-03, -4.0683156e-04, ...,\n","                       1.5679072e-03, -1.2422163e-03, -1.4097716e-03],\n","                     [-1.0383492e-04,  6.2583291e-05, -3.1657887e-04, ...,\n","                      -1.1479309e-03,  2.2410708e-03,  1.1157807e-03],\n","                     [-1.0741745e-03,  1.7077600e-03, -2.1457406e-04, ...,\n","                       2.1380140e-03,  2.0076728e-03, -1.7538114e-03],\n","                     ...,\n","                     [ 1.4712225e-04, -8.8894728e-04,  2.0345475e-03, ...,\n","                       1.3683115e-04,  2.7806012e-04,  2.4470135e-03],\n","                     [-5.9876987e-04, -3.4406217e-04, -1.3000756e-03, ...,\n","                       1.7987278e-04, -1.0222688e-03,  7.9509732e-04],\n","                     [-1.2763214e-03, -9.2914613e-04,  5.6984997e-04, ...,\n","                       2.4844732e-03, -8.3725573e-04, -1.8913789e-03]], dtype=float32),\n","              array([ 3.9678981e-04, -8.4300729e-04, -4.1830914e-05, ...,\n","                     -4.3345310e-04, -6.9622579e-04,  5.2373158e-04], dtype=float32)),\n","             (array([[ 6.7763485e-04,  1.0260126e-03, -3.2743039e-05, ...,\n","                       9.9753251e-04, -9.3875767e-04, -5.4222305e-04],\n","                     [-5.3303968e-04, -5.2766164e-04, -2.8118477e-04, ...,\n","                       8.2746614e-04, -7.5240969e-05, -1.1203006e-03],\n","                     [ 1.9715377e-04, -1.3271766e-03,  6.8825677e-05, ...,\n","                      -1.2165394e-03, -9.9351874e-04,  6.2603754e-04],\n","                     ...,\n","                     [ 7.2531786e-04, -6.6834484e-04,  2.9620223e-04, ...,\n","                      -1.5554606e-05,  4.5569488e-04, -9.5159312e-06],\n","                     [ 3.9901776e-04,  1.2041263e-03,  3.2373003e-04, ...,\n","                      -7.9501071e-04,  5.3982815e-04,  4.0282929e-04],\n","                     [ 8.6060099e-05, -9.2329108e-04,  1.5056246e-03, ...,\n","                       1.2389227e-03,  4.5052395e-05,  1.7775111e-03]], dtype=float32),\n","              array([-0.00035745, -0.00020203, -0.00034553, ...,  0.00043496,\n","                      0.00026238,  0.000162  ], dtype=float32)),\n","             (array([[-6.5632758e-04,  1.9307630e-04, -2.3937158e-04, ...,\n","                      -2.9769450e-04, -3.9326059e-04, -3.3545130e-04],\n","                     [ 3.4865465e-05, -5.6964450e-04,  8.4580861e-05, ...,\n","                       7.0200826e-04, -1.5650771e-04, -9.9007064e-04],\n","                     [-8.1516788e-05, -1.6839606e-04,  1.8943823e-04, ...,\n","                       1.8949553e-04,  1.1072367e-04,  1.1003779e-04],\n","                     ...,\n","                     [-5.2142768e-05,  4.8017919e-05, -4.3895040e-04, ...,\n","                      -5.8524776e-04,  9.5390942e-04,  4.1829702e-04],\n","                     [-3.9770694e-05, -3.8482141e-04,  4.4685442e-04, ...,\n","                      -2.7321349e-04, -1.0609677e-04,  8.1117230e-04],\n","                     [-6.1787153e-04,  1.1259716e-04, -2.7352778e-04, ...,\n","                       4.7451831e-04, -4.7100461e-04,  4.9875362e-04]], dtype=float32),\n","              array([-5.0122751e-05,  1.3252236e-04,  4.4255292e-05, ...,\n","                     -1.6132134e-04, -1.1098751e-05,  7.2615258e-05], dtype=float32))),\n","            (),\n","            (array([[ 1.70283718e-04, -4.41001641e-04,  7.63004500e-05, ...,\n","                     -1.08118504e-04, -4.23405785e-04, -5.40672117e-05],\n","                    [-3.94014991e-04,  5.54013648e-04,  5.15887630e-04, ...,\n","                      7.62916825e-05,  4.33876121e-05,  2.54852202e-04],\n","                    [ 6.70946611e-04, -1.61401374e-04, -2.70279794e-04, ...,\n","                     -2.46808086e-05, -6.28456473e-04,  1.14964350e-04],\n","                    ...,\n","                    [ 3.92061222e-04, -7.70126178e-04,  3.81032660e-05, ...,\n","                     -6.01870488e-05,  2.15118969e-04, -4.70766354e-05],\n","                    [ 1.45630867e-04, -3.02121116e-05, -2.34290666e-04, ...,\n","                     -8.99372462e-05, -3.68241454e-04, -1.11075133e-04],\n","                    [-2.70546472e-04,  7.31936540e-04,  7.86951714e-05, ...,\n","                     -7.80509363e-05, -3.87212815e-04, -9.37166798e-04]], dtype=float32),\n","             array([-6.1937702e-05, -6.7385423e-05, -3.6533346e-04, ...,\n","                    -3.8991460e-05, -1.4449641e-04, -5.3948129e-04], dtype=float32))), ())),\n","         ()),\n","        ()))),\n","     ()),\n","    (((),\n","      ((),\n","       ((array([ 1.9965204,  7.917583 , 17.497572 , ...,  5.580989 ,  9.620596 ,\n","                15.242364 ], dtype=float32),\n","         array([-0.000487  ,  0.00040629, -0.00430074, ..., -0.00025506,\n","                 0.00068512, -0.000792  ], dtype=float32)),\n","        (array([[ 7.81930925e-04,  5.36141852e-05, -5.24379313e-04, ...,\n","                  6.19613566e-05, -1.14736764e-03,  7.18329320e-05],\n","                [-6.35195291e-04,  5.03429255e-05,  5.31960162e-04, ...,\n","                 -4.76386427e-04,  3.31802003e-04,  3.90495930e-04],\n","                [ 2.52242549e-04,  8.99954830e-05,  5.38007123e-04, ...,\n","                  3.07680166e-04,  3.32221010e-04,  1.58135241e-04],\n","                ...,\n","                [ 4.40601179e-05,  3.81713558e-04,  1.00201760e-04, ...,\n","                  5.59606240e-04, -6.36014884e-05,  3.52097704e-04],\n","                [-1.90641527e-04, -1.04702103e-05, -6.71170768e-04, ...,\n","                 -8.23360111e-04, -2.29277852e-04, -2.62569345e-04],\n","                [ 2.74527149e-04, -6.56365941e-04, -4.61192132e-04, ...,\n","                  1.09481414e-04, -6.97781288e-05,  4.79980983e-04]], dtype=float32),\n","         array([-0.0001663 , -0.0002241 , -0.00016707, ..., -0.00038479,\n","                -0.00029605, -0.00016414], dtype=float32)),\n","        ((), (), ()),\n","        (),\n","        (array([[-8.1292912e-04,  1.2577352e-03,  6.2077510e-04, ...,\n","                  2.9063295e-04, -2.8643024e-04, -3.5083198e-04],\n","                [-8.6263288e-04, -3.0233312e-04, -1.9231575e-06, ...,\n","                 -6.9119284e-05,  2.3940593e-04,  1.4589172e-04],\n","                [ 3.4782017e-04,  3.7570691e-04, -1.4653345e-04, ...,\n","                 -3.3968252e-05,  1.3544984e-04, -9.7487718e-05],\n","                ...,\n","                [-3.0106737e-04,  3.6231737e-04, -2.7745136e-04, ...,\n","                  1.2673953e-04, -3.2690249e-04, -3.8023651e-04],\n","                [ 4.2692645e-04, -5.6375331e-05, -8.2404214e-05, ...,\n","                 -7.5063208e-04,  6.6593348e-05,  1.6361207e-04],\n","                [-4.4173099e-05, -1.1191863e-03,  7.6636250e-05, ...,\n","                 -1.0756662e-04,  6.3053821e-04, -4.5731175e-04]], dtype=float32),\n","         array([-1.5893763e-04,  6.6568050e-06,  5.1825651e-04, ...,\n","                -1.8336775e-04,  2.3202836e-05, -1.2177654e-04], dtype=float32)),\n","        ()))),\n","     ()),\n","    (((),\n","      ((), ((array([ 6.6538224 ,  5.8633456 , -0.02628019, ...,  5.235887  ,\n","                 5.315903  ,  6.1849303 ], dtype=float32),\n","         array([-5.7357101e-04,  1.1880817e-04,  4.3790329e-05, ...,\n","                 7.3501957e-05, -3.0689713e-04,  1.1272966e-04], dtype=float32)),\n","        ((),\n","         ((),\n","          ((), (((array([[-1.4202745e-03, -6.4122357e-04, -3.1149792e-03, ...,\n","                       6.8108871e-04,  1.0233371e-04, -2.5829396e-04],\n","                     [ 1.1793796e-03,  1.1446518e-03,  2.9955430e-05, ...,\n","                      -1.4046218e-03,  1.6074018e-03,  1.2349782e-03],\n","                     [ 1.1469198e-03, -4.6678263e-04, -6.2566943e-04, ...,\n","                       9.2009688e-04, -5.7193241e-04, -8.9929666e-04],\n","                     ...,\n","                     [-6.0266734e-04, -2.8245908e-05, -8.4985321e-04, ...,\n","                      -6.2882621e-04,  6.7810446e-04, -4.0915958e-04],\n","                     [-5.2926684e-05, -7.1604096e-04,  5.7407329e-04, ...,\n","                       5.1823299e-04, -2.5810942e-04, -1.9388416e-04],\n","                     [ 5.3689966e-04, -7.2543410e-04,  3.0148085e-04, ...,\n","                      -2.3811084e-04,  2.1353961e-04,  6.3435500e-04]], dtype=float32),\n","              array([-0.00215339,  0.00258201,  0.0016408 , ..., -0.00145559,\n","                      0.00102888, -0.00015717], dtype=float32)),\n","             (array([[-3.5809506e-05, -1.5991062e-03, -1.3092887e-03, ...,\n","                       1.4074763e-03, -4.8352740e-04, -6.4611662e-04],\n","                     [ 5.6862691e-04, -9.0099109e-04,  7.6649711e-04, ...,\n","                      -3.3085706e-04, -2.7406358e-04,  1.0744719e-03],\n","                     [ 5.5510836e-04, -4.5776035e-04,  1.2680170e-03, ...,\n","                      -9.7430064e-05, -1.0577477e-04,  3.4940205e-04],\n","                     ...,\n","                     [-1.0253923e-03, -2.1394966e-04, -4.2553892e-04, ...,\n","                      -1.0938757e-03,  1.3573157e-03,  1.5445183e-04],\n","                     [-3.9241856e-04,  4.4109780e-04,  7.5339293e-04, ...,\n","                      -4.7016842e-04,  1.1076404e-03, -1.6160314e-03],\n","                     [-7.2982610e-04,  3.7579660e-04,  3.5862732e-04, ...,\n","                      -1.4565125e-03,  3.3435581e-04,  3.8607532e-04]], dtype=float32),\n","              array([ 8.7614098e-05, -5.6735618e-04,  1.3058398e-04, ...,\n","                     -8.3180363e-05, -4.4650737e-05,  2.4007722e-04], dtype=float32)),\n","             (array([[ 3.2658994e-04, -8.9762878e-05, -3.0128303e-04, ...,\n","                       2.0412290e-04,  2.1946381e-04, -2.8178160e-04],\n","                     [-1.0100439e-03,  1.0037237e-03,  3.4821985e-04, ...,\n","                      -8.9144282e-04, -3.2818769e-04, -4.3555719e-05],\n","                     [ 1.3798120e-04, -1.9146658e-04,  9.1560214e-05, ...,\n","                       5.0999130e-05, -2.6068137e-05, -7.1690365e-06],\n","                     ...,\n","                     [-1.7983539e-04,  6.3701838e-05, -4.4978373e-05, ...,\n","                       1.9647813e-04, -2.0324369e-04,  4.7924306e-04],\n","                     [ 1.6455063e-04,  3.3378717e-04, -4.2719787e-04, ...,\n","                      -2.8767748e-04,  3.3329992e-04,  6.5964885e-04],\n","                     [ 5.9143844e-04, -1.0050208e-05, -8.3477302e-05, ...,\n","                       8.0683705e-04, -4.4600657e-04,  8.1023732e-05]], dtype=float32),\n","              array([-6.41180350e-06, -3.73999137e-06, -1.51658634e-04, ...,\n","                      4.28922031e-05, -1.15473915e-04, -1.11402202e-04], dtype=float32))),\n","            (),\n","            (array([[-3.8550294e-04,  2.1398910e-04,  3.2397365e-04, ...,\n","                     -2.7557669e-04, -3.9520729e-04,  1.8977985e-04],\n","                    [ 4.3297734e-04, -2.9576322e-04, -6.4963382e-04, ...,\n","                     -5.0657097e-04, -4.7725014e-04, -1.6216185e-05],\n","                    [-2.5697506e-04, -3.9215005e-04, -2.1434038e-04, ...,\n","                      4.5999966e-04,  7.0850179e-04, -4.4175319e-04],\n","                    ...,\n","                    [ 2.5891548e-04,  4.4266495e-04,  1.0796422e-04, ...,\n","                     -3.8441244e-04, -1.5695074e-05, -2.8410493e-04],\n","                    [-2.4241036e-04, -1.1702699e-05,  2.3457655e-04, ...,\n","                     -2.6948302e-04, -6.1991380e-04,  7.1968971e-05],\n","                    [ 2.0332196e-04, -4.2470350e-04, -3.8880802e-04, ...,\n","                     -7.7080447e-04, -4.6517767e-04, -1.3774162e-04]], dtype=float32),\n","             array([ 4.0583360e-05, -7.7631921e-05, -2.6093237e-04, ...,\n","                    -7.1511444e-05,  1.2536180e-04, -7.8641846e-05], dtype=float32))), ())),\n","         ()),\n","        ()))),\n","     ()),\n","    (((),\n","      ((), ((array([ 0.86686385, 12.802946  , 20.918468  , ...,  8.096811  ,\n","                11.982407  , 21.11511   ], dtype=float32),\n","         array([-2.8954102e-05, -3.9707223e-04, -3.2913175e-03, ...,\n","                 3.8375551e-04,  6.1936228e-04, -4.7001746e-04], dtype=float32)),\n","        (array([[-3.8925868e-05, -2.6646629e-04,  2.6528424e-04, ...,\n","                 -4.3724454e-04,  3.6093232e-05,  6.7226548e-04],\n","                [-1.6310242e-04, -1.3488269e-04, -9.0547313e-04, ...,\n","                 -1.7097283e-05, -2.4584580e-05, -3.9190726e-04],\n","                [ 1.2887633e-04,  4.1654348e-04,  5.9360737e-04, ...,\n","                  3.1710719e-04,  3.3515587e-04,  4.7137315e-04],\n","                ...,\n","                [-8.4278564e-04,  3.8631633e-04,  2.9718073e-04, ...,\n","                 -7.5707689e-04,  4.9767736e-04, -5.7628484e-05],\n","                [-4.4372078e-04, -5.0279620e-04,  1.1439355e-04, ...,\n","                 -7.0099963e-04, -5.3970388e-04,  4.4876995e-04],\n","                [ 4.5658104e-04, -4.6831122e-04,  1.0177303e-03, ...,\n","                  3.6785172e-04, -2.2921641e-04, -2.7541595e-04]], dtype=float32),\n","         array([-0.00021363, -0.00017123, -0.00022285, ..., -0.00019269,\n","                -0.00035768, -0.00017541], dtype=float32)),\n","        ((), (), ()),\n","        (),\n","        (array([[ 4.03159793e-04, -6.84049737e-05, -5.86503884e-04, ...,\n","                 -2.96753336e-04, -5.66961360e-04, -4.33594134e-04],\n","                [ 1.94050444e-04, -4.59901930e-04, -3.67373228e-04, ...,\n","                 -6.82132435e-04,  5.27080265e-04, -8.98202343e-05],\n","                [ 1.11393434e-04, -4.42333490e-04,  8.91838048e-04, ...,\n","                  7.72579981e-04, -1.30207918e-04, -8.52136465e-04],\n","                ...,\n","                [ 5.49096789e-04, -4.92376334e-04, -6.00103638e-04, ...,\n","                 -4.16681665e-04,  2.03793068e-04, -1.53737143e-04],\n","                [-4.09635337e-04,  4.58542374e-04, -3.39150749e-04, ...,\n","                  2.89692598e-05,  2.99449050e-04,  4.16583905e-04],\n","                [ 7.49631145e-05, -5.85675647e-04, -4.18101670e-04, ...,\n","                  1.47251179e-04, -5.53502003e-04,  1.49513967e-03]], dtype=float32),\n","         array([-6.9049529e-05, -7.4342647e-06,  4.9019692e-04, ...,\n","                -1.1256799e-04, -6.6731343e-05,  1.4789050e-04], dtype=float32)),\n","        ()))),\n","     ()),\n","    (((),\n","      ((),\n","       ((array([13.884472 , 13.461945 ,  2.5632052, ..., 12.236495 , 15.360047 ,\n","                13.01401  ], dtype=float32),\n","         array([-4.55720990e-04,  2.17849956e-04, -1.40621298e-04, ...,\n","                 1.20455996e-04, -8.02556024e-05, -4.55700683e-05], dtype=float32)),\n","        ((),\n","         ((),\n","          ((),\n","           (((array([[ 3.04347104e-05,  4.59795876e-04, -3.58563848e-04, ...,\n","                      -6.67466156e-06,  2.14641984e-03, -1.00333348e-03],\n","                     [ 5.95779158e-04,  1.14064384e-03, -1.07206719e-03, ...,\n","                      -4.66909521e-04,  7.73672218e-05, -9.38461511e-04],\n","                     [ 9.38303361e-04, -1.52905029e-03, -4.02868638e-04, ...,\n","                      -4.83391574e-04,  1.03019185e-04, -1.41213904e-03],\n","                     ...,\n","                     [-5.64618909e-04, -1.09574059e-03,  5.65421768e-04, ...,\n","                      -1.32750755e-03,  1.51475938e-03, -1.50439853e-03],\n","                     [-1.15687086e-04,  5.31895435e-04, -2.54979241e-04, ...,\n","                       4.83278185e-04, -3.56908015e-04,  9.57636395e-04],\n","                     [-8.46514478e-04, -1.67643547e-03, -1.20342374e-04, ...,\n","                      -2.46390817e-04,  1.19053386e-03, -1.06173800e-03]], dtype=float32),\n","              array([-0.00031459,  0.00080458,  0.00039606, ...,  0.00015511,\n","                     -0.00010833,  0.00056067], dtype=float32)),\n","             (array([[-9.6203832e-05, -7.5434516e-05, -1.3908293e-05, ...,\n","                      -7.5202831e-04,  6.0623442e-04,  2.9144905e-04],\n","                     [-7.9408271e-04,  4.7140758e-04, -1.9731325e-04, ...,\n","                       1.3953816e-04, -1.2602318e-03,  1.1233679e-03],\n","                     [ 8.0137671e-04,  5.5163377e-04,  2.6442393e-04, ...,\n","                      -4.4417346e-04,  2.0911905e-04,  1.4202296e-03],\n","                     ...,\n","                     [ 2.2483942e-04, -6.1800657e-04,  3.5468265e-04, ...,\n","                      -2.7550667e-04, -9.7560161e-04, -1.3231493e-03],\n","                     [ 5.0081260e-04,  4.7377041e-05, -2.5712102e-04, ...,\n","                      -7.6909031e-04, -1.5295447e-03,  6.8073496e-05],\n","                     [-4.4313041e-04, -3.5339271e-04, -8.2671277e-06, ...,\n","                       1.3777794e-03,  4.4238818e-04, -2.6601285e-04]], dtype=float32),\n","              array([-4.0534799e-04, -1.8297087e-05, -4.1196903e-04, ...,\n","                      1.8859327e-04, -2.2690948e-04,  1.9148938e-04], dtype=float32)),\n","             (array([[-3.6900310e-05,  5.3377364e-05, -3.1843616e-04, ...,\n","                       5.0405116e-04, -2.6211084e-04,  1.3316567e-04],\n","                     [ 2.7131135e-04,  3.0249747e-04,  4.0872645e-05, ...,\n","                      -7.6816732e-04, -3.1976314e-04, -1.2593270e-04],\n","                     [-3.1932891e-06,  8.7763865e-06, -1.4936092e-05, ...,\n","                      -5.3743708e-05,  4.0223610e-04,  1.2810763e-04],\n","                     ...,\n","                     [-6.0343988e-05,  3.8131751e-04, -5.0654786e-04, ...,\n","                      -3.0548431e-04,  7.8835146e-05, -3.4251998e-04],\n","                     [-2.9731393e-04, -7.5186741e-05,  1.5473201e-04, ...,\n","                      -2.8104385e-04, -7.9728459e-05, -2.9720634e-04],\n","                     [-3.2589093e-04,  2.5464472e-04,  4.2908837e-04, ...,\n","                      -2.5658301e-04,  8.4630016e-04,  8.0578197e-05]], dtype=float32),\n","              array([ 1.31171664e-05, -1.16424606e-04, -4.70318955e-05, ...,\n","                     -2.68741260e-05, -7.25425271e-05,  9.96946437e-06], dtype=float32))),\n","            (),\n","            (array([[-2.95275007e-04,  5.61425462e-04,  1.94941633e-04, ...,\n","                      1.67324339e-04,  1.83305950e-04,  3.90420144e-04],\n","                    [-2.60702247e-04, -7.89239057e-05, -1.91575909e-05, ...,\n","                     -1.73649081e-04,  8.05039090e-05, -9.14192788e-05],\n","                    [ 6.23444794e-05,  5.79497835e-04,  2.60153291e-04, ...,\n","                     -3.75368807e-04, -1.56709531e-04, -2.89433199e-04],\n","                    ...,\n","                    [-2.81711924e-04,  3.98267963e-04,  1.18682881e-04, ...,\n","                     -4.78833652e-04,  4.10300418e-04,  5.47525531e-04],\n","                    [-3.62313549e-05,  1.67650811e-04,  2.34106032e-04, ...,\n","                      2.63145601e-04,  2.44905561e-04, -1.04096264e-03],\n","                    [-3.14409292e-04, -2.86370108e-04,  7.87048048e-05, ...,\n","                      3.84655956e-04, -1.06067404e-04,  4.90543025e-04]], dtype=float32),\n","             array([ 4.0300263e-05,  2.3128554e-04, -4.0615199e-04, ...,\n","                    -1.0913893e-04,  7.5368938e-05,  1.6266171e-05], dtype=float32))),\n","           ())),\n","         ()),\n","        ()))),\n","     ()),\n","    (((),\n","      ((), ((array([ 0.90813017, 13.274897  , 27.00867   , ..., 10.683291  ,\n","                14.673738  , 22.71965   ], dtype=float32),\n","         array([-0.00040027,  0.00080252, -0.00560015, ..., -0.00020622,\n","                 0.00074947, -0.00134835], dtype=float32)),\n","        (array([[ 4.2194486e-05, -2.0589048e-04,  1.6432903e-05, ...,\n","                  8.2122732e-04, -1.7064164e-04,  3.0993717e-04],\n","                [-5.3452142e-04,  4.9766062e-05, -3.0989287e-04, ...,\n","                 -5.5295194e-04, -1.0995404e-04, -3.0539738e-04],\n","                [ 1.6639604e-04,  6.4287364e-04,  2.1608412e-04, ...,\n","                  6.3951901e-04,  4.4746877e-04,  6.2048831e-04],\n","                ...,\n","                [-2.7214037e-04, -4.1171981e-04, -4.9864844e-04, ...,\n","                  7.8231731e-04, -3.8248757e-04,  6.6872622e-04],\n","                [ 1.5704014e-04, -3.0455180e-04, -5.2021455e-04, ...,\n","                 -5.4276467e-04,  7.0829602e-04,  2.2111159e-04],\n","                [-1.0274536e-03, -4.1698874e-04,  1.6419195e-04, ...,\n","                  1.3148717e-06, -4.7315675e-04,  2.6419692e-04]], dtype=float32),\n","         array([-0.00015792, -0.0002012 , -0.00040425, ..., -0.00029917,\n","                -0.00026957, -0.00029693], dtype=float32)),\n","        ((), (), ()),\n","        (),\n","        (array([[-4.87683516e-04,  4.60849085e-04, -7.67914287e-04, ...,\n","                 -2.94859899e-04,  5.63545291e-05, -5.63330890e-04],\n","                [-3.40761151e-04,  1.10351839e-05, -7.62750045e-04, ...,\n","                 -1.24265876e-04, -4.14661452e-04, -8.41802903e-05],\n","                [-1.11599744e-04,  1.70214422e-04, -1.51115790e-04, ...,\n","                  1.80554955e-04, -3.73218529e-04, -1.59875344e-04],\n","                ...,\n","                [ 2.34882245e-04,  3.95135343e-04, -4.64887766e-04, ...,\n","                  1.11324407e-05, -8.93000251e-05, -1.67855207e-04],\n","                [-2.19646899e-04,  1.01316918e-03, -4.98502166e-04, ...,\n","                 -3.43046762e-04, -2.59156077e-04, -5.11897728e-04],\n","                [-6.27335277e-04,  2.79073371e-04, -6.19138940e-04, ...,\n","                  1.23680802e-04, -7.06370454e-04, -4.92858700e-04]], dtype=float32),\n","         array([ 7.7389333e-05, -1.3871471e-04,  7.0207054e-04, ...,\n","                -1.0981207e-04, -5.8232330e-05,  1.2948278e-04], dtype=float32)),\n","        ()))),\n","     ()),\n","    (((),\n","      ((), ((array([ 8.372775  , 15.97151   ,  0.18343599, ..., 12.533916  ,\n","                13.81566   , 11.615755  ], dtype=float32),\n","         array([-6.1237760e-04,  1.1887442e-05,  2.8334666e-04, ...,\n","                 1.5710128e-04,  8.6874446e-05,  4.1886221e-04], dtype=float32)),\n","        ((),\n","         ((),\n","          ((),\n","           (((array([[-5.44616953e-04, -2.51836784e-04,  2.82613473e-06, ...,\n","                       6.86037703e-04, -2.15495049e-04,  2.44984840e-04],\n","                     [ 3.02632281e-04,  1.52682909e-03, -4.72705782e-04, ...,\n","                      -9.84038808e-04,  1.07976620e-03,  1.07629132e-03],\n","                     [ 3.96817573e-04, -1.20164550e-04, -7.33872759e-04, ...,\n","                      -5.05906588e-04,  1.09366840e-03,  7.17497373e-04],\n","                     ...,\n","                     [-3.23485874e-04,  3.53168522e-04,  2.73831654e-04, ...,\n","                      -1.14989933e-03,  7.51305313e-04,  1.46175630e-03],\n","                     [ 1.18979915e-04,  1.33986723e-05, -2.24436808e-04, ...,\n","                       5.20060363e-04, -1.20040076e-03,  3.97292868e-04],\n","                     [-5.09415288e-04, -6.32965995e-04,  3.10948089e-04, ...,\n","                       3.53982876e-04,  8.25071707e-04,  1.05326879e-03]], dtype=float32),\n","              array([-3.8780083e-04, -8.8340741e-05,  2.4179350e-04, ...,\n","                      2.4433064e-04, -1.3791240e-03, -6.3623063e-04], dtype=float32)),\n","             (array([[ 3.63493833e-04,  2.22630861e-05,  4.13535046e-04, ...,\n","                       2.67355954e-05,  1.14081590e-03, -3.03737961e-05],\n","                     [-6.12370437e-04,  3.85504391e-04, -8.79621191e-04, ...,\n","                       8.13836596e-05, -1.19204051e-03,  7.95744185e-04],\n","                     [-6.95833733e-05, -1.57878821e-04,  6.42060244e-04, ...,\n","                       6.79816236e-04, -6.38233265e-04, -2.90615018e-04],\n","                     ...,\n","                     [ 5.12468068e-05,  2.33415223e-04,  2.18864268e-04, ...,\n","                      -1.95501780e-04, -1.03498045e-04,  1.71859434e-03],\n","                     [-1.89528582e-05, -2.27364708e-05, -4.30362532e-04, ...,\n","                       1.14310568e-03, -1.23944308e-03, -8.22421047e-04],\n","                     [ 1.80473944e-04, -2.02382042e-04,  5.59071312e-04, ...,\n","                      -2.47066746e-05,  1.10022281e-03, -6.45871391e-04]], dtype=float32),\n","              array([-7.5146738e-05, -6.7579836e-06,  1.4482529e-04, ...,\n","                     -8.1040795e-05,  2.1821246e-04, -2.3070970e-05], dtype=float32)),\n","             (array([[ 3.05953399e-05, -9.24778360e-05,  8.66919348e-04, ...,\n","                       5.02216630e-04,  1.47341416e-04,  3.39233171e-04],\n","                     [ 4.36710361e-05,  4.76982794e-04, -4.73783643e-04, ...,\n","                       3.07363720e-04, -1.15720075e-04,  7.83116993e-05],\n","                     [ 1.17371164e-04,  1.12441012e-05, -7.10471431e-05, ...,\n","                       1.41282741e-04,  2.01951189e-04,  6.06731737e-05],\n","                     ...,\n","                     [ 4.40982927e-04, -2.52492377e-04,  7.50637264e-05, ...,\n","                      -1.61263481e-04, -2.84032401e-04, -3.83365405e-04],\n","                     [ 1.20402001e-04,  4.33679350e-04, -1.77962836e-04, ...,\n","                       4.97163157e-04, -5.35714098e-05, -4.14361857e-04],\n","                     [-1.44370657e-04, -1.23632286e-04,  3.02367058e-04, ...,\n","                      -3.64843087e-04,  4.54966910e-04, -1.85745812e-04]], dtype=float32),\n","              array([-1.3675290e-04,  6.3294588e-05, -8.0753256e-05, ...,\n","                     -3.2648273e-05,  3.1480609e-04, -6.0649669e-05], dtype=float32))),\n","            (),\n","            (array([[-1.63617078e-04,  3.67685076e-04,  1.98180991e-04, ...,\n","                     -2.40701353e-04, -5.89832634e-05, -5.30336001e-05],\n","                    [ 2.54022802e-04, -4.25478269e-04, -3.20644795e-05, ...,\n","                      2.88880343e-04, -5.38854292e-05,  1.82518066e-04],\n","                    [-1.16012219e-04,  3.34622135e-04,  1.12896996e-04, ...,\n","                      4.48528444e-05,  6.73738599e-04,  8.79551590e-05],\n","                    ...,\n","                    [-2.25719603e-04, -3.26944282e-04, -2.69738637e-04, ...,\n","                     -3.02909291e-04,  8.59866559e-04,  5.53370628e-04],\n","                    [ 1.34181770e-04, -6.57585100e-04, -9.94502770e-05, ...,\n","                     -6.21918880e-04,  3.47616995e-04, -4.17777774e-04],\n","                    [ 3.06540402e-04, -6.03168337e-05,  4.69886872e-04, ...,\n","                     -4.88033576e-04,  1.24226935e-04, -5.76646707e-05]], dtype=float32),\n","             array([ 0.00019565, -0.00014278, -0.00022448, ..., -0.00012007,\n","                     0.00014003, -0.00017284], dtype=float32))),\n","           ())),\n","         ()),\n","        ()))),\n","     ()),\n","    (((),\n","      ((), ((array([-0.06025542, 12.744205  , 25.656012  , ..., 10.303179  ,\n","                15.503817  , 24.314907  ], dtype=float32),\n","         array([-1.4057662e-04,  8.3549810e-04, -2.9817980e-03, ...,\n","                -3.5448023e-04, -8.4938816e-05, -4.9499440e-04], dtype=float32)),\n","        (array([[ 1.7180972e-04, -2.3920540e-04,  7.5348729e-04, ...,\n","                 -1.3515765e-04,  1.0145999e-04,  1.8247469e-05],\n","                [-1.4432312e-04, -1.1674406e-03, -3.9437515e-04, ...,\n","                 -1.7912679e-04, -3.5602535e-04,  5.6255516e-04],\n","                [ 2.9295412e-04, -8.6514548e-05,  5.7367206e-04, ...,\n","                  4.3877316e-04,  3.3841399e-04,  2.1447285e-04],\n","                ...,\n","                [ 2.2061571e-05, -3.2794542e-04,  6.9817237e-04, ...,\n","                 -1.4935243e-04, -6.5445079e-04, -1.8960223e-04],\n","                [ 1.7179751e-04, -8.6728920e-04, -4.2238689e-04, ...,\n","                 -2.9146395e-04, -1.0349107e-03,  1.5651011e-04],\n","                [ 1.2128260e-05, -2.5433020e-04,  2.5173649e-04, ...,\n","                 -4.7979589e-05,  4.6818659e-05, -2.8438959e-05]], dtype=float32),\n","         array([-3.4387287e-04, -5.0625644e-05, -1.6881192e-04, ...,\n","                -1.7907440e-04, -2.9558962e-04, -1.2355539e-04], dtype=float32)),\n","        ((), (), ()),\n","        (),\n","        (array([[ 3.6168432e-05,  6.1548554e-04,  3.0334532e-04, ...,\n","                  4.8002921e-04,  5.5033244e-05,  2.8048758e-04],\n","                [-5.9433543e-05, -5.8846100e-04, -3.5217364e-04, ...,\n","                 -2.5843770e-05, -3.0673057e-04, -6.4706532e-05],\n","                [-2.8164592e-04,  2.3700466e-04, -4.1411966e-04, ...,\n","                  3.1590275e-04,  4.1643166e-04,  1.3534189e-04],\n","                ...,\n","                [ 3.1502338e-04,  1.1009246e-03, -2.8655128e-04, ...,\n","                 -3.2970747e-05,  4.6942721e-04, -9.9466264e-04],\n","                [-9.8968165e-05,  5.9347722e-04,  1.1509573e-05, ...,\n","                 -8.0399757e-04, -6.2405644e-04, -5.9072394e-04],\n","                [-1.3721759e-04, -3.6486497e-04, -1.6709921e-04, ...,\n","                 -2.2322961e-04,  3.3870182e-04, -1.7233458e-04]], dtype=float32),\n","         array([ 1.71015912e-04, -1.39437898e-05,  8.43976391e-04, ...,\n","                -1.16422860e-04,  1.07966725e-05,  3.21390224e-04], dtype=float32)),\n","        ()))),\n","     ()),\n","    (((),\n","      ((),\n","       ((array([14.994992 , 19.581324 ,  0.4059196, ..., 17.31174  , 20.743263 ,\n","                17.49876  ], dtype=float32),\n","         array([ 4.1027187e-04, -6.9329733e-05,  1.4972858e-03, ...,\n","                 7.5663927e-05, -2.0776749e-04,  3.9869780e-04], dtype=float32)),\n","        ((),\n","         ((),\n","          ((), (((array([[ 9.1546314e-04,  1.5524508e-03,  2.5012778e-04, ...,\n","                      -4.9519946e-04,  5.7675538e-04, -4.4260957e-04],\n","                     [-7.4331224e-04, -4.5568543e-04, -1.0788902e-04, ...,\n","                      -3.2766731e-04, -5.0470342e-05,  6.1477348e-04],\n","                     [-1.6673241e-04, -4.8464135e-04,  6.5355987e-04, ...,\n","                       1.2788856e-03, -4.3500238e-04,  1.0445452e-03],\n","                     ...,\n","                     [ 2.1309736e-03,  1.6212454e-03,  4.1409006e-04, ...,\n","                      -7.8390894e-04,  2.3823480e-04, -4.5184096e-04],\n","                     [-2.3623992e-04, -2.6248675e-04,  8.4920292e-04, ...,\n","                      -5.8198045e-04,  5.1756209e-04, -8.3849410e-04],\n","                     [-1.0358334e-03,  3.4658294e-04, -1.7187942e-03, ...,\n","                      -3.5548038e-04, -2.1329139e-04,  6.3356834e-05]], dtype=float32),\n","              array([ 0.0004397 , -0.00020229, -0.00014382, ..., -0.0004822 ,\n","                      0.0004883 , -0.00041549], dtype=float32)),\n","             (array([[-2.3763950e-04, -2.7378398e-04, -4.5901322e-04, ...,\n","                       5.5614999e-04, -4.1898157e-04,  4.9971812e-04],\n","                     [ 9.1988310e-05,  6.7285197e-05, -1.6225618e-04, ...,\n","                       5.6705798e-04, -5.2078150e-04,  1.2263547e-04],\n","                     [ 4.4181594e-04,  3.8721444e-04,  4.8599509e-04, ...,\n","                       8.2972139e-04, -8.1753277e-04,  2.9794374e-04],\n","                     ...,\n","                     [-7.2789303e-04, -6.9660746e-04, -5.3178257e-04, ...,\n","                       4.7599318e-05, -2.6572464e-04,  3.1057670e-04],\n","                     [-1.1280369e-05, -3.6910738e-04,  3.5289800e-04, ...,\n","                       8.5157859e-07, -1.9613527e-04, -3.0006384e-04],\n","                     [-2.1678836e-03, -1.2203567e-03, -2.4673119e-04, ...,\n","                      -2.2490606e-04,  2.8064745e-04, -7.0838578e-04]], dtype=float32),\n","              array([-5.5110781e-04, -7.3397718e-04, -9.4174058e-04, ...,\n","                      6.5562286e-05,  3.0734044e-04, -6.8180911e-05], dtype=float32)),\n","             (array([[-3.37552570e-04, -1.03581304e-04,  5.88525028e-04, ...,\n","                      -2.66055984e-04,  3.96084215e-04,  5.92434488e-04],\n","                     [-3.37541365e-04, -3.56916498e-05, -5.31644291e-05, ...,\n","                       2.48546712e-04,  1.79222683e-04, -1.59390678e-04],\n","                     [ 8.17786786e-05, -1.11311892e-04, -1.61984848e-04, ...,\n","                      -5.12819970e-04, -2.25731550e-04,  6.60209480e-05],\n","                     ...,\n","                     [-4.10632812e-04, -5.26667805e-04, -3.61651910e-04, ...,\n","                      -3.52680130e-04, -4.20080469e-05, -3.49334558e-04],\n","                     [-2.30420017e-04,  7.62987882e-04,  4.46458464e-04, ...,\n","                      -4.38721418e-06,  3.85945395e-06,  3.08426475e-04],\n","                     [-6.01766107e-04, -8.35633909e-05,  3.44379747e-04, ...,\n","                      -4.73200285e-04, -6.36932789e-04, -2.21948329e-04]], dtype=float32),\n","              array([ 1.1037237e-05, -1.0091416e-04,  3.4272354e-05, ...,\n","                     -4.0069423e-05, -3.3001867e-05,  2.9715884e-05], dtype=float32))),\n","            (),\n","            (array([[ 5.1698234e-04,  1.8031315e-04, -3.8759818e-04, ...,\n","                     -3.1627095e-04,  2.0707610e-04,  5.8595015e-04],\n","                    [ 4.5675650e-04, -1.5976759e-04, -7.4758609e-05, ...,\n","                      1.3149201e-04, -2.7493079e-04,  1.4753541e-04],\n","                    [-1.3478673e-03, -6.3602172e-04,  5.6097732e-04, ...,\n","                      1.3072652e-04, -5.2393833e-04, -1.6751020e-04],\n","                    ...,\n","                    [ 2.0112176e-04, -4.2648194e-04,  1.8004706e-04, ...,\n","                      7.2160110e-05, -2.4541170e-05,  3.8859044e-04],\n","                    [-3.4597132e-04, -4.2280437e-05, -4.2710005e-04, ...,\n","                     -3.7438040e-05,  9.6285810e-05,  2.7012386e-04],\n","                    [-3.3104874e-04,  1.9421008e-04, -8.7771514e-05, ...,\n","                      1.5493046e-04, -3.7556578e-04,  2.1193750e-04]], dtype=float32),\n","             array([-8.5814230e-05, -1.9713718e-05, -4.6342684e-04, ...,\n","                     6.2028128e-05,  4.5731522e-05, -1.4597982e-05], dtype=float32))), ())),\n","         ()),\n","        ()))),\n","     ()),\n","    (((),\n","      ((),\n","       ((array([1.6655147e-02, 1.6082397e+01, 3.1873774e+01, ..., 1.1174616e+01,\n","                1.7513676e+01, 2.9441385e+01], dtype=float32),\n","         array([-2.3196573e-05,  1.7523501e-03, -8.4896814e-03, ...,\n","                 1.6299316e-04,  1.5865959e-04, -3.0854838e-03], dtype=float32)),\n","        (array([[ 3.15710669e-04, -5.15163818e-04,  5.61184599e-04, ...,\n","                  8.63516179e-05,  5.67984942e-04, -1.05282561e-04],\n","                [-1.19735945e-04,  2.34216132e-05, -3.19164392e-04, ...,\n","                 -1.55681177e-04,  2.31536309e-04, -4.07057261e-04],\n","                [ 9.52258706e-04,  2.10977843e-04,  8.85121772e-05, ...,\n","                  1.56357986e-04,  1.36118877e-04,  4.64385550e-04],\n","                ...,\n","                [-1.40872726e-04, -7.28056766e-04,  9.62022823e-05, ...,\n","                 -1.86030113e-04, -6.80152953e-05, -4.74927831e-04],\n","                [ 2.85371614e-04,  6.20496343e-04, -7.36928705e-05, ...,\n","                 -2.59382155e-04,  9.56919539e-05,  8.42265399e-06],\n","                [-3.58539459e-04,  6.32089505e-04,  1.27168641e-06, ...,\n","                 -7.19042553e-04, -1.75549503e-04,  2.84050948e-05]], dtype=float32),\n","         array([-1.5254301e-04, -3.7107213e-05, -1.6394313e-04, ...,\n","                -1.3167717e-04, -2.4609649e-04, -3.2261750e-04], dtype=float32)),\n","        ((), (), ()),\n","        (),\n","        (array([[ 6.06400368e-04,  1.92615887e-04, -6.83535996e-04, ...,\n","                  1.72328975e-04,  3.59775207e-04, -5.54463651e-04],\n","                [-3.42816405e-04, -1.21603407e-04, -8.56918414e-05, ...,\n","                 -2.66241899e-04, -1.07160384e-04,  5.95166639e-04],\n","                [ 2.53094418e-04,  1.77691385e-04, -4.50531574e-04, ...,\n","                  3.76148091e-04,  2.35175292e-04,  3.51075752e-04],\n","                ...,\n","                [ 2.93284975e-04, -3.78099474e-04,  2.38093024e-04, ...,\n","                 -3.57929646e-04, -7.25283695e-04, -6.94682967e-05],\n","                [ 8.71256780e-05,  5.59157750e-04, -3.95087234e-04, ...,\n","                 -5.65570081e-04,  3.91754247e-06, -1.68332481e-04],\n","                [ 1.68647632e-04, -2.57659649e-05, -6.86358107e-05, ...,\n","                  5.61154040e-04, -1.45072991e-04, -1.11228263e-04]], dtype=float32),\n","         array([ 3.6376034e-06, -2.2723810e-04,  8.7604468e-04, ...,\n","                 4.3749093e-05,  6.6529661e-05,  4.7133752e-04], dtype=float32)),\n","        ()))),\n","     ()),\n","    (((),\n","      ((), ((array([16.788107, 22.658192,  8.172792, ..., 20.255539, 24.943554,\n","                26.92817 ], dtype=float32),\n","         array([-1.2250307e-04, -3.2226005e-04, -7.0443912e-04, ...,\n","                 7.3070056e-05,  3.2760476e-04,  5.5552536e-04], dtype=float32)),\n","        ((),\n","         ((),\n","          ((), (((array([[ 1.2876471e-03,  1.0384790e-03, -3.7831510e-04, ...,\n","                       7.1057654e-04, -2.3340413e-04,  5.6387583e-04],\n","                     [ 2.1036977e-03,  1.7847327e-04, -1.8438653e-03, ...,\n","                       6.0670856e-05,  3.7204962e-05,  8.0581398e-05],\n","                     [-1.3639807e-03,  4.8973074e-04,  7.0214545e-04, ...,\n","                       5.4646924e-04, -2.8453162e-04, -1.6602727e-03],\n","                     ...,\n","                     [ 5.0478254e-04, -5.9482845e-04, -1.2755810e-03, ...,\n","                      -3.8480351e-04,  9.4723381e-04, -5.2645011e-04],\n","                     [ 8.2717033e-04, -2.6428993e-04, -8.6759881e-04, ...,\n","                      -1.4107502e-03, -6.1278709e-04, -3.8625213e-04],\n","                     [-1.0050079e-03,  2.1177696e-04, -2.1596905e-03, ...,\n","                      -4.3031588e-04, -1.2404476e-03, -5.8588904e-04]], dtype=float32),\n","              array([ 0.0010137 , -0.00048201, -0.00073613, ..., -0.00012195,\n","                      0.0003789 ,  0.0009197 ], dtype=float32)),\n","             (array([[-1.4257472e-04, -2.2734975e-04,  4.9845030e-04, ...,\n","                       7.5627968e-04, -7.7000679e-04, -5.8322353e-06],\n","                     [-1.0113433e-03,  5.7670090e-04, -1.7385422e-03, ...,\n","                       1.1266858e-03,  1.5786490e-04,  5.8013224e-04],\n","                     [ 6.7959911e-05, -2.0278418e-04, -3.2667324e-04, ...,\n","                       6.2981642e-05,  6.4648438e-04, -2.5587686e-05],\n","                     ...,\n","                     [-8.0522691e-04, -2.6556561e-04, -6.4855511e-04, ...,\n","                      -8.0774218e-05,  1.9374531e-04,  1.1016447e-03],\n","                     [-4.0460125e-04,  8.6580066e-04, -8.1822992e-04, ...,\n","                       6.4003200e-04, -8.1704668e-04,  9.8370037e-05],\n","                     [-3.2770375e-05,  2.0451602e-04,  1.4611923e-04, ...,\n","                      -3.2215912e-04,  6.9541699e-04,  1.8413069e-03]], dtype=float32),\n","              array([-0.00032509, -0.00030064,  0.00024112, ..., -0.0001434 ,\n","                      0.00015936, -0.00010945], dtype=float32)),\n","             (array([[-2.56667903e-04,  1.35117676e-04,  2.48232129e-04, ...,\n","                      -3.37227335e-04,  1.15631075e-04, -6.03324967e-04],\n","                     [ 2.24190953e-04, -2.97536579e-04,  3.92053946e-04, ...,\n","                      -5.10960293e-04, -8.50013457e-04,  2.49507924e-04],\n","                     [-3.27645888e-04,  3.70969967e-04, -2.83983682e-04, ...,\n","                      -1.02944854e-04,  1.64038604e-04, -2.49171746e-04],\n","                     ...,\n","                     [-3.75199998e-05, -2.27302240e-04,  5.96545462e-04, ...,\n","                      -4.64063487e-05, -4.48283885e-04, -3.56353761e-04],\n","                     [ 3.61789571e-04,  1.15724186e-04,  5.26799977e-06, ...,\n","                      -6.97995943e-04,  8.97933569e-05, -4.00072779e-04],\n","                     [-9.80078130e-06, -8.17500040e-06,  5.19030669e-04, ...,\n","                      -1.21871293e-04,  7.41585798e-04, -5.18545567e-04]], dtype=float32),\n","              array([-1.3537475e-04,  2.9878329e-06,  3.7983109e-04, ...,\n","                     -7.2705181e-05,  2.8456616e-06, -3.2132706e-05], dtype=float32))),\n","            (),\n","            (array([[ 8.4341055e-04,  5.8063352e-04,  1.1073208e-04, ...,\n","                     -8.3404793e-06,  6.5528101e-04,  3.4908947e-04],\n","                    [ 3.8023677e-06, -1.5541370e-04,  1.0752342e-04, ...,\n","                     -3.3008278e-04, -3.6879609e-04, -2.1309481e-04],\n","                    [-1.2604462e-04,  7.5949298e-05,  8.6022919e-06, ...,\n","                      3.1998329e-06,  4.2465510e-04,  7.4662914e-04],\n","                    ...,\n","                    [-3.6020670e-04,  8.1935636e-04, -1.0574513e-04, ...,\n","                     -5.5663777e-04,  6.5840525e-04, -2.2332350e-04],\n","                    [-3.7615467e-04, -3.2222229e-05,  3.4371897e-05, ...,\n","                      4.0855442e-04,  5.6322780e-04, -2.1649050e-04],\n","                    [ 7.8592607e-04,  3.3197875e-04,  1.8338383e-04, ...,\n","                     -2.1694263e-04,  8.5363333e-04,  9.2418968e-06]], dtype=float32),\n","             array([-9.8701326e-05,  2.8400600e-04, -4.2735788e-04, ...,\n","                     6.3747444e-05, -1.2107831e-06, -5.9743377e-04], dtype=float32))), ())),\n","         ()),\n","        ()))),\n","     ()),\n","    (((),\n","      ((), ((array([-0.22309317, 13.583151  , 34.711605  , ..., 17.613558  ,\n","                18.067015  , 31.221674  ], dtype=float32),\n","         array([-1.0765874e-03,  1.2495470e-03, -7.5413720e-03, ...,\n","                -2.6556994e-05,  1.8623150e-04, -2.6852870e-03], dtype=float32)),\n","        (array([[-2.7290365e-04, -1.3811997e-04, -1.6225104e-05, ...,\n","                  1.7597360e-04,  3.7111051e-04, -2.2392054e-04],\n","                [ 4.9703848e-04,  3.7176782e-04, -2.5990239e-04, ...,\n","                  7.4534719e-06,  4.6654246e-04, -8.8737998e-04],\n","                [ 3.3166632e-04,  7.4939866e-04,  3.5576944e-04, ...,\n","                  5.5805006e-04,  2.4689225e-04,  5.6866166e-04],\n","                ...,\n","                [-4.9539180e-05, -6.0123118e-04, -2.3322008e-04, ...,\n","                  8.4160612e-04,  2.1532476e-04,  2.5216991e-04],\n","                [ 4.6039798e-04, -3.4750180e-04, -8.5163378e-04, ...,\n","                 -6.6180440e-04, -4.5377237e-04, -6.7425624e-04],\n","                [ 1.1132313e-04, -1.8683717e-05, -1.9954798e-04, ...,\n","                  1.2992531e-05, -3.3742897e-04,  5.0666835e-04]], dtype=float32),\n","         array([-0.00036083, -0.00033533, -0.00035955, ..., -0.00022611,\n","                -0.00014071, -0.00035067], dtype=float32)),\n","        ((), (), ()),\n","        (),\n","        (array([[-1.17857053e-04, -1.45043105e-05,  4.62007156e-04, ...,\n","                 -4.58927207e-05,  2.95146048e-04,  6.10764138e-04],\n","                [-5.74538950e-04, -5.15882057e-05, -1.90355655e-04, ...,\n","                  3.73110524e-05, -6.64523162e-04, -1.10983336e-03],\n","                [ 5.09286649e-04,  1.22181955e-03,  7.68135767e-04, ...,\n","                  6.13221433e-04, -7.33257039e-04, -3.78598634e-04],\n","                ...,\n","                [ 1.95730288e-04, -3.50462687e-05,  8.31331345e-05, ...,\n","                  4.86384262e-04, -3.80867103e-04,  1.22060796e-04],\n","                [ 1.47242550e-04,  1.63499513e-04, -6.04007102e-04, ...,\n","                  8.96162310e-05,  4.82854346e-04,  2.31110869e-04],\n","                [ 2.73201033e-04, -8.58253276e-04, -6.17735321e-04, ...,\n","                 -4.82715346e-04, -5.10380232e-05, -2.58517364e-04]], dtype=float32),\n","         array([-5.9670725e-05, -2.9360230e-05,  7.8188692e-04, ...,\n","                -1.2617938e-05,  1.8969568e-04,  3.9778469e-04], dtype=float32)),\n","        ()))),\n","     ()),\n","    (((),\n","      ((), ((array([12.76066 , 24.670063,  7.782917, ..., 21.25663 , 25.444324,\n","                23.580734], dtype=float32),\n","         array([-4.7950973e-04, -3.7119153e-05, -6.1405625e-04, ...,\n","                 1.0852494e-04,  2.5389714e-05,  4.8603112e-04], dtype=float32)),\n","        ((),\n","         ((),\n","          ((), (((array([[-1.8794144e-03, -4.0391227e-04, -7.4245851e-05, ...,\n","                       4.1299048e-04, -1.5075152e-03, -1.0696204e-03],\n","                     [-2.4199445e-04, -4.4900476e-04,  1.0185906e-03, ...,\n","                      -9.8011841e-04,  8.9059520e-04, -5.7093048e-04],\n","                     [-2.0954767e-04, -9.7880349e-04, -1.6399295e-03, ...,\n","                      -1.9985903e-03, -1.0594971e-03,  1.6108782e-03],\n","                     ...,\n","                     [-7.7370182e-04,  8.8151445e-04, -3.1441305e-04, ...,\n","                      -1.2598165e-03, -2.1084132e-04, -2.3244745e-04],\n","                     [-1.3420199e-03, -8.2608924e-04,  8.6183113e-04, ...,\n","                       2.1711301e-07,  1.8767304e-04,  9.8143096e-07],\n","                     [ 1.7102381e-04, -3.1594725e-04,  3.5516551e-04, ...,\n","                      -2.5578635e-03, -7.2237098e-04, -1.8428755e-04]], dtype=float32),\n","              array([ 6.5682598e-06,  1.9866459e-04,  5.1888457e-04, ...,\n","                      7.5079972e-04,  4.4039168e-04, -6.3585542e-04], dtype=float32)),\n","             (array([[-6.3174398e-04, -2.3356917e-04, -5.0414109e-04, ...,\n","                       1.0674710e-03, -3.0254214e-03, -5.9655076e-04],\n","                     [ 1.5039378e-03,  7.0800784e-04, -1.9573972e-04, ...,\n","                      -8.3176896e-04, -4.7667953e-04, -1.3188428e-04],\n","                     [ 3.0009035e-04,  2.0620439e-04,  5.2325981e-04, ...,\n","                       1.1599436e-03, -3.4125766e-04,  9.5831172e-04],\n","                     ...,\n","                     [ 9.5937372e-04, -4.6781065e-06, -1.1564007e-03, ...,\n","                      -4.3456681e-04, -3.1979443e-04, -1.1943850e-03],\n","                     [-1.3989415e-04,  5.2804913e-04,  3.6480196e-05, ...,\n","                      -4.6509830e-04,  4.4058191e-04, -1.1029021e-03],\n","                     [ 3.2544121e-04, -3.9745330e-05,  5.0852820e-04, ...,\n","                      -3.1518890e-04,  9.1476068e-05,  2.0392031e-04]], dtype=float32),\n","              array([ 2.6584388e-04,  3.3843812e-06,  2.9595123e-04, ...,\n","                      4.4296711e-04,  5.4846157e-04, -3.9269114e-05], dtype=float32)),\n","             (array([[ 9.17628058e-05, -2.28567718e-04,  1.81682786e-04, ...,\n","                      -5.78264648e-04,  2.81282002e-04, -7.98615565e-06],\n","                     [ 1.65738777e-04,  5.60371380e-04,  2.70049914e-05, ...,\n","                       1.01919723e-04, -1.37445517e-04,  1.25345628e-04],\n","                     [ 1.84871838e-04,  2.08280660e-04,  1.09634901e-04, ...,\n","                       1.18935626e-04, -1.70320825e-04,  1.53050540e-04],\n","                     ...,\n","                     [ 4.75964800e-04,  2.20564281e-04, -3.47592504e-05, ...,\n","                      -3.28224269e-04, -4.41108423e-04,  2.07057281e-04],\n","                     [-8.23619601e-04,  1.55788055e-03,  2.02693907e-03, ...,\n","                       2.50819343e-04, -4.15284390e-04,  9.98374599e-05],\n","                     [ 2.57274456e-04, -8.96865444e-04,  3.71781498e-05, ...,\n","                      -1.58791197e-04, -3.09414609e-04,  7.55404064e-04]], dtype=float32),\n","              array([-3.6296603e-07, -7.3279647e-05, -6.3085710e-05, ...,\n","                     -7.6242686e-05,  8.7460823e-05,  3.0841152e-06], dtype=float32))),\n","            (),\n","            (array([[-2.54634942e-04, -2.03693256e-04,  6.05449779e-04, ...,\n","                     -4.52605134e-04,  4.02736390e-04, -1.17780895e-04],\n","                    [-1.78130285e-05,  3.25516303e-04, -1.89110951e-03, ...,\n","                      2.59718618e-05, -3.39911116e-04, -7.72585627e-04],\n","                    [ 2.96866754e-04, -6.16304169e-04, -5.03267802e-04, ...,\n","                     -1.08384556e-04, -8.66749906e-06,  8.49635864e-04],\n","                    ...,\n","                    [ 3.53985350e-04,  5.32949518e-04, -6.41448540e-04, ...,\n","                     -1.68419851e-04,  3.79585632e-04, -3.17466365e-05],\n","                    [ 3.59581696e-04,  1.01122388e-03,  6.58348377e-04, ...,\n","                     -1.70839470e-04, -2.67660856e-04,  4.00556455e-05],\n","                    [ 4.40707139e-04, -8.95877369e-04, -2.68703268e-04, ...,\n","                      2.65317445e-04,  3.76862431e-06, -1.54678200e-04]], dtype=float32),\n","             array([ 7.6193930e-05,  1.1616956e-04, -4.5604061e-04, ...,\n","                    -6.6918146e-05, -1.9460129e-05, -2.2153524e-04], dtype=float32))), ())),\n","         ()),\n","        ()))),\n","     ()),\n","    (((),\n","      ((),\n","       ((array([ 1.0337392, 22.061754 , 43.231194 , ..., 19.429026 , 23.564987 ,\n","                38.006355 ], dtype=float32),\n","         array([-0.00117999,  0.00206893, -0.00709111, ..., -0.00124938,\n","                -0.00020696, -0.00268608], dtype=float32)),\n","        (array([[-2.01435556e-04, -6.91106077e-04,  3.14379460e-04, ...,\n","                  1.76711750e-04,  3.50645045e-04,  7.80623086e-05],\n","                [-3.67916917e-04,  3.66886932e-04, -7.24520127e-04, ...,\n","                 -4.00645658e-04, -4.16928262e-04, -1.00789075e-05],\n","                [ 5.43883478e-04,  2.92949146e-04,  4.60255018e-04, ...,\n","                  3.96303003e-05,  6.81075209e-04,  7.74378248e-04],\n","                ...,\n","                [-7.68236350e-05, -2.92517041e-04, -3.18741804e-04, ...,\n","                  4.21491801e-04, -9.30934330e-04, -5.57764200e-04],\n","                [ 1.61084317e-04, -2.74029444e-04, -1.57639679e-05, ...,\n","                 -2.54638464e-04, -3.29936505e-04, -1.01829661e-04],\n","                [ 3.10885196e-04, -3.35270946e-04,  4.54016663e-06, ...,\n","                 -6.03569555e-04,  1.86554767e-04,  4.05824161e-04]], dtype=float32),\n","         array([-0.0001799 , -0.00030168, -0.0002745 , ..., -0.000102  ,\n","                -0.00034411, -0.00033481], dtype=float32)),\n","        ((), (), ()),\n","        (),\n","        (array([[ 5.8286131e-04,  6.6324201e-04, -3.1632103e-04, ...,\n","                 -1.0869616e-04, -5.3245132e-04, -4.9036207e-05],\n","                [ 2.3049131e-04,  2.3240889e-05, -2.1291108e-04, ...,\n","                  1.3178992e-04,  3.9156678e-04,  1.4489120e-05],\n","                [ 2.3855596e-04,  5.3774694e-04,  1.0776903e-04, ...,\n","                 -2.0306613e-04,  1.0866073e-04, -6.6243467e-04],\n","                ...,\n","                [ 5.1894051e-04,  5.9537904e-04, -5.7935249e-04, ...,\n","                 -4.0819831e-04, -6.2351450e-05, -1.3086008e-03],\n","                [-4.9656729e-04, -3.5608186e-06,  9.7273849e-05, ...,\n","                 -6.3281687e-04,  3.4597132e-04,  3.3747064e-04],\n","                [-9.5058087e-05,  1.1434412e-03,  6.1862745e-05, ...,\n","                  1.9864331e-04, -5.4002373e-04,  7.8538396e-05]], dtype=float32),\n","         array([ 8.6805230e-05, -1.8472261e-04,  9.4641180e-04, ...,\n","                -7.1869617e-05,  6.2471117e-05,  5.9154339e-04], dtype=float32)),\n","        ()))),\n","     ()),\n","    (((),\n","      ((), ((array([ 5.308498  ,  9.052232  , -0.03863969, ...,  7.6905227 ,\n","                 9.103197  ,  7.572106  ], dtype=float32),\n","         array([-3.568901e-04, -4.437698e-04,  5.805128e-04, ..., -8.461932e-05,\n","                 7.270499e-05,  3.239243e-04], dtype=float32)),\n","        ((),\n","         ((),\n","          ((),\n","           (((array([[-1.30673812e-04,  6.59130048e-04,  7.80374248e-05, ...,\n","                       5.23762312e-04, -8.70782067e-04, -8.73823068e-04],\n","                     [ 4.32819361e-04, -1.69462830e-04,  3.51712486e-04, ...,\n","                       7.17637769e-04,  1.06508459e-03,  8.85349931e-04],\n","                     [ 4.30888293e-04,  5.77962201e-04, -2.00252354e-04, ...,\n","                      -1.53001354e-04, -3.19540239e-04,  1.05178144e-04],\n","                     ...,\n","                     [-4.88329097e-04, -4.30004788e-04, -8.41957342e-04, ...,\n","                      -9.84652084e-04, -7.99078902e-04, -9.04035056e-04],\n","                     [-4.73572145e-04,  5.24294388e-04,  7.42647564e-04, ...,\n","                      -1.56144422e-04, -1.02069185e-04, -1.01500067e-04],\n","                     [-7.79024049e-05,  7.58415787e-04,  8.67914176e-04, ...,\n","                       7.94223510e-04,  1.83557824e-03,  1.14687078e-03]], dtype=float32),\n","              array([-9.7116642e-04,  3.1526803e-04,  7.8963378e-04, ...,\n","                     -7.3155170e-05, -2.7635656e-04, -4.0302941e-04], dtype=float32)),\n","             (array([[ 1.3081268e-03,  2.6335689e-04,  1.0288443e-03, ...,\n","                       1.2157984e-03,  1.6164865e-03,  3.7799717e-04],\n","                     [ 2.9952067e-04, -1.1680925e-03, -5.2662421e-04, ...,\n","                      -7.3047311e-05, -3.6435091e-04, -7.6652557e-04],\n","                     [-9.1706211e-04,  3.2160673e-04,  1.5769685e-03, ...,\n","                       5.3916010e-04,  2.0885631e-04,  3.8118521e-04],\n","                     ...,\n","                     [-5.0397532e-04,  2.2079983e-04, -3.5761227e-04, ...,\n","                       2.6838994e-04,  2.6896439e-04,  2.4520911e-04],\n","                     [ 4.9634173e-04, -1.4896430e-04,  5.8065396e-04, ...,\n","                       7.3420459e-05,  4.4524484e-05,  5.1391256e-05],\n","                     [-4.9626695e-05,  1.1330956e-03, -4.7166621e-05, ...,\n","                       2.2529240e-04, -3.6124414e-04, -3.9860918e-04]], dtype=float32),\n","              array([ 3.0119956e-04, -1.3478730e-04, -2.6088383e-04, ...,\n","                     -6.2855188e-06,  1.6467554e-04,  1.2802162e-04], dtype=float32)),\n","             (array([[-2.96418730e-04, -6.00164094e-05,  3.00254032e-04, ...,\n","                       3.63416533e-04,  1.25651262e-04,  9.12271717e-05],\n","                     [-7.36662769e-04, -1.11273410e-04,  4.34085552e-04, ...,\n","                      -4.85641387e-04,  3.29294780e-05,  1.69852574e-04],\n","                     [-3.81841091e-05,  9.87484382e-05,  1.00617137e-04, ...,\n","                      -2.74354461e-05, -7.97469547e-05,  7.18402498e-06],\n","                     ...,\n","                     [-5.19409252e-04,  1.13916458e-05,  4.14684037e-04, ...,\n","                       7.18128940e-05, -2.27463126e-04,  1.37359413e-04],\n","                     [-1.63358145e-05,  1.05720965e-05, -9.19181184e-05, ...,\n","                      -3.66109220e-04,  5.15847118e-04, -3.62631632e-04],\n","                     [-3.97869444e-04,  6.50472748e-06, -1.42097299e-04, ...,\n","                       3.67718079e-04, -5.30160651e-05, -5.57318737e-04]], dtype=float32),\n","              array([ 2.0855365e-04,  1.6737364e-04, -1.0294465e-04, ...,\n","                     -5.6050238e-05, -2.1637079e-05,  3.0401079e-05], dtype=float32))),\n","            (),\n","            (array([[ 2.9206814e-04,  5.3977204e-04, -8.2436782e-05, ...,\n","                      2.2202826e-04, -7.2031347e-05, -2.2803305e-04],\n","                    [ 6.6677731e-04,  1.0669035e-04,  3.5072511e-04, ...,\n","                     -4.1412302e-05,  1.9790417e-04, -4.6948122e-04],\n","                    [-6.4396317e-04, -1.6291814e-04,  3.7853143e-04, ...,\n","                     -1.4875385e-04, -1.1362734e-04,  7.9966901e-04],\n","                    ...,\n","                    [ 1.2799914e-04,  3.3890683e-04, -5.5582990e-04, ...,\n","                     -2.2343966e-05,  2.4609949e-04,  5.0496624e-04],\n","                    [ 2.9816595e-04, -2.1838253e-04,  3.3140945e-04, ...,\n","                      1.4986371e-04, -3.3596077e-04, -4.2342530e-05],\n","                    [-2.3466033e-04, -5.0972856e-04,  6.0179108e-04, ...,\n","                     -2.5697795e-04,  9.1924921e-05, -5.9286319e-04]], dtype=float32),\n","             array([ 2.0774532e-04, -1.2198475e-06, -2.2798477e-04, ...,\n","                    -9.4303512e-05, -3.9003335e-05, -2.8486713e-04], dtype=float32))),\n","           ())),\n","         ()),\n","        ()))),\n","     ()),\n","    (((),\n","      ((),\n","       ((array([ 2.0158238, 23.0998   , 48.96083  , ..., 17.659014 , 28.608997 ,\n","                41.778034 ], dtype=float32),\n","         array([-7.8206591e-05,  2.5509086e-03, -6.1729825e-03, ...,\n","                -1.0399139e-03, -3.1973262e-04, -2.3189490e-03], dtype=float32)),\n","        (array([[ 1.16511335e-04, -1.15627990e-04,  1.67550781e-04, ...,\n","                 -5.08037163e-04, -2.30883052e-05,  4.54896828e-04],\n","                [ 2.60431734e-05, -1.20925535e-04,  2.17910652e-04, ...,\n","                 -9.62103513e-05, -1.48444029e-04, -8.78057035e-04],\n","                [ 5.55114006e-04,  1.18374919e-04,  1.48948311e-04, ...,\n","                  1.90981096e-04,  5.88726718e-04, -7.15185161e-05],\n","                ...,\n","                [-3.83822451e-04,  3.95453011e-04,  1.72172586e-04, ...,\n","                  7.02342077e-05, -5.69404336e-04, -2.63927504e-04],\n","                [-5.24173840e-04,  2.23404604e-05,  2.50776444e-04, ...,\n","                 -6.04471134e-04, -4.94876411e-04,  3.25881294e-04],\n","                [-2.65447597e-05,  3.71686338e-06, -4.86109464e-04, ...,\n","                 -1.41719298e-04,  1.58686407e-06,  7.32640619e-04]], dtype=float32),\n","         array([-4.5249160e-04, -2.6979944e-04, -5.2003961e-05, ...,\n","                -2.6564312e-04, -2.1713240e-04, -7.2526971e-05], dtype=float32)),\n","        ((), (), ()),\n","        (),\n","        (array([[-5.4372872e-06,  2.8050537e-04, -5.2871817e-04, ...,\n","                 -2.2510379e-04, -1.8546477e-04,  8.0518934e-05],\n","                [-1.0688491e-05, -8.5127953e-04, -3.4959632e-04, ...,\n","                  4.2957367e-04,  9.2931099e-05,  1.3921224e-04],\n","                [ 1.3818906e-04, -1.8654544e-04,  7.2355353e-04, ...,\n","                  9.1411086e-05, -8.2753053e-05, -3.0220038e-04],\n","                ...,\n","                [-1.7757314e-04,  5.5445166e-04, -3.8052309e-05, ...,\n","                  4.2014729e-04, -8.9350040e-04,  5.1400799e-04],\n","                [-1.1905436e-04, -3.8618618e-04, -3.6971227e-04, ...,\n","                 -4.6468168e-04, -9.6176867e-05, -1.2701953e-05],\n","                [ 2.8910377e-04,  2.7694437e-04,  6.1447034e-04, ...,\n","                 -2.6935749e-04, -5.2958349e-04,  7.2338281e-04]], dtype=float32),\n","         array([ 8.3583378e-05, -3.7047378e-04,  1.0382327e-03, ...,\n","                 6.0955454e-06,  1.6792382e-04,  6.5312092e-04], dtype=float32)),\n","        ()))),\n","     ()),\n","    (((),\n","      ((),\n","       ((array([18.206758 , 26.892624 ,  3.9487524, ..., 24.221392 , 27.340975 ,\n","                20.403324 ], dtype=float32),\n","         array([ 2.9363800e-04, -3.0046096e-04,  5.9346861e-04, ...,\n","                 1.4421754e-04, -1.8973135e-05,  5.7468051e-04], dtype=float32)),\n","        ((),\n","         ((),\n","          ((), (((array([[ 4.4347992e-04, -1.1271415e-03, -4.7793113e-05, ...,\n","                       6.1281963e-04, -6.4743089e-04,  4.6867484e-04],\n","                     [ 8.5233088e-04,  3.3860229e-04, -2.9998302e-04, ...,\n","                      -9.0523608e-05, -1.2703490e-03, -1.7373421e-04],\n","                     [-1.9600100e-03,  1.0812135e-03, -1.6841863e-03, ...,\n","                       1.6678463e-03,  6.3569564e-04, -3.5963556e-05],\n","                     ...,\n","                     [-3.1490711e-04, -1.2447604e-03, -8.1193744e-04, ...,\n","                       6.0631143e-04, -8.9656637e-04,  1.2479685e-03],\n","                     [ 2.3099859e-04, -3.5308304e-04,  5.8214861e-04, ...,\n","                       2.6479107e-04,  6.8508773e-05,  5.3705407e-05],\n","                     [-1.8215231e-05,  5.3064903e-04, -1.0810395e-03, ...,\n","                       1.0328778e-03,  4.9995666e-04,  1.2128409e-03]], dtype=float32),\n","              array([ 6.1109208e-04, -4.9780076e-04,  7.7879283e-04, ...,\n","                     -7.5881433e-04, -5.8737251e-06, -1.7239941e-04], dtype=float32)),\n","             (array([[-7.5738452e-04,  3.4037212e-04,  9.4604737e-05, ...,\n","                      -1.9614406e-04,  1.1578086e-03,  5.4415478e-04],\n","                     [-8.2472153e-04,  1.4717685e-04,  5.9594185e-04, ...,\n","                       5.3338427e-04,  3.2996168e-04,  3.1371214e-04],\n","                     [-1.0635330e-03, -1.1787401e-03, -1.9173555e-04, ...,\n","                      -2.4164771e-03,  1.3534717e-03, -7.9612114e-04],\n","                     ...,\n","                     [-3.4121479e-04,  7.8552280e-04,  3.3117947e-04, ...,\n","                      -8.8593457e-04,  1.0311465e-04,  5.0930929e-04],\n","                     [-5.6616828e-04,  8.3911233e-04,  3.5107546e-04, ...,\n","                      -3.4310299e-04, -4.3623158e-04, -5.8756757e-04],\n","                     [-2.0634252e-04, -2.6424133e-04, -1.7309857e-03, ...,\n","                      -6.1521085e-04, -1.3453382e-03, -4.2951101e-04]], dtype=float32),\n","              array([-1.2847008e-04, -8.3963590e-05, -1.5528119e-04, ...,\n","                     -3.6382617e-04,  2.3820289e-04, -7.1337663e-06], dtype=float32)),\n","             (array([[-6.9462396e-05, -1.2945353e-04,  2.4404624e-04, ...,\n","                      -5.0651963e-04, -7.5763650e-04, -3.1681691e-04],\n","                     [-5.3991675e-07,  1.5787637e-05,  9.3035640e-05, ...,\n","                       1.7101782e-04,  9.7495649e-05, -1.6427528e-04],\n","                     [ 3.2023006e-04, -1.5589209e-04,  1.4839959e-04, ...,\n","                      -4.2911110e-04,  8.7888424e-05,  4.3223794e-05],\n","                     ...,\n","                     [ 4.8685056e-04, -2.2934660e-04, -2.7203135e-04, ...,\n","                       1.5696247e-04, -4.9142953e-05, -1.4183215e-04],\n","                     [ 5.0828883e-05,  1.9128654e-04,  1.1916073e-05, ...,\n","                      -1.9904935e-04, -4.7654932e-04,  4.3682163e-04],\n","                     [-1.5550673e-04, -2.1200035e-04, -3.2963351e-04, ...,\n","                       1.2730011e-04,  1.5033255e-04, -3.6193596e-04]], dtype=float32),\n","              array([ 1.5241512e-05,  1.4000875e-06, -1.2576176e-04, ...,\n","                      1.6966784e-04, -4.6086672e-05, -9.9027704e-05], dtype=float32))),\n","            (),\n","            (array([[-3.9493789e-05, -1.1709694e-04, -1.8582463e-04, ...,\n","                      1.2467768e-04, -2.2823880e-04,  1.8728113e-04],\n","                    [-3.2068708e-04,  2.1949438e-04,  8.2432643e-07, ...,\n","                     -3.4469721e-04,  1.3793644e-04, -2.2019669e-04],\n","                    [-8.6381049e-05,  3.4842742e-04,  1.7045446e-04, ...,\n","                     -3.6513302e-04, -6.3988281e-04,  2.2804254e-04],\n","                    ...,\n","                    [ 5.7569885e-04, -5.2908767e-04,  2.4378620e-04, ...,\n","                      2.6649068e-04,  3.0112744e-04, -2.1738380e-04],\n","                    [ 6.7063351e-04, -4.2031502e-04,  2.0448044e-04, ...,\n","                      2.0350120e-04,  3.0722728e-04,  4.7925278e-05],\n","                    [ 1.6859004e-04,  5.7874894e-04, -8.1104646e-04, ...,\n","                     -7.6212914e-04, -1.5044890e-04, -4.1823620e-05]], dtype=float32),\n","             array([-1.51491215e-04, -1.46320854e-05, -7.13975693e-04, ...,\n","                     1.20466306e-04,  5.48123062e-05, -2.06700017e-04], dtype=float32))), ())),\n","         ()),\n","        ()))),\n","     ()),\n","    (((),\n","      ((), ((array([ 7.191653, 29.466003, 55.35567 , ..., 25.048405, 31.487995,\n","                50.610085], dtype=float32),\n","         array([-0.00137473,  0.00311482, -0.01180465, ..., -0.00171093,\n","                 0.00069104, -0.00538259], dtype=float32)),\n","        (array([[-2.2953753e-04,  3.3690110e-05,  1.5238048e-04, ...,\n","                  3.3377192e-04,  4.3300868e-04,  9.9708734e-05],\n","                [-1.5647816e-04, -2.4447052e-05, -8.1881102e-05, ...,\n","                  1.9484751e-04,  1.7551563e-04, -3.1832376e-04],\n","                [ 4.8789737e-04,  1.9329342e-04,  5.5228797e-04, ...,\n","                  2.6952982e-04,  9.6460548e-04,  8.0640096e-04],\n","                ...,\n","                [-7.1828451e-04,  3.0392030e-05,  2.6537199e-04, ...,\n","                 -1.4989957e-04,  8.9847122e-04,  4.9114984e-04],\n","                [-4.3935256e-04, -2.6561355e-04,  3.3806649e-04, ...,\n","                  4.3255271e-04, -5.6894106e-04,  2.8261484e-04],\n","                [ 4.2059502e-04,  5.0217455e-04, -4.7125132e-04, ...,\n","                 -3.1130802e-04,  2.1810968e-04,  5.5348495e-04]], dtype=float32),\n","         array([-0.0002097 , -0.00018928, -0.00024592, ..., -0.00012104,\n","                -0.00016394, -0.00034142], dtype=float32)),\n","        ((), (), ()),\n","        (),\n","        (array([[ 5.9382062e-05, -8.6231514e-05, -6.0964130e-06, ...,\n","                  3.8866565e-04,  1.7018507e-04,  2.5088439e-04],\n","                [-3.9774194e-04, -4.2988028e-04,  7.8191020e-05, ...,\n","                  4.6574202e-04, -5.2762404e-04, -1.0509451e-04],\n","                [ 3.6966222e-04,  1.7330467e-05, -5.4959144e-04, ...,\n","                  5.0344839e-05, -7.8359834e-04,  4.3859785e-05],\n","                ...,\n","                [ 4.9921399e-04,  1.4461572e-03, -4.2671917e-04, ...,\n","                  1.9091791e-04, -7.6773152e-04, -6.8730209e-04],\n","                [ 3.1187490e-04, -5.0244551e-05,  6.5637473e-04, ...,\n","                  1.3351483e-04,  7.7946531e-04, -9.2227740e-05],\n","                [-6.0682814e-04,  2.5083861e-04,  1.1597313e-04, ...,\n","                  2.4455061e-04, -3.8216342e-04, -1.4542563e-04]], dtype=float32),\n","         array([-4.5304550e-05, -3.6065068e-04,  9.0119435e-04, ...,\n","                 7.6837685e-05,  1.2223882e-04,  8.2975440e-04], dtype=float32)),\n","        ()))),\n","     ()),\n","    (((),\n","      ((), ((array([ 8.354686, 16.813698,  8.461904, ..., 18.02643 , 21.150185,\n","                14.28129 ], dtype=float32),\n","         array([-0.00052863, -0.00018351, -0.00131444, ...,  0.00016475,\n","                 0.00014666,  0.0003576 ], dtype=float32)),\n","        ((),\n","         ((),\n","          ((), (((array([[ 9.8654116e-04, -4.1079312e-04,  5.8836642e-05, ...,\n","                       1.2607159e-03,  5.7120435e-04, -2.4733468e-04],\n","                     [ 8.5880002e-04,  2.1833180e-04, -1.2253980e-03, ...,\n","                       8.1404654e-04, -6.8213901e-04,  5.2649726e-04],\n","                     [-1.1412364e-03,  1.3554128e-03,  1.8227930e-03, ...,\n","                      -3.4494500e-03,  2.0212708e-04, -9.9548209e-04],\n","                     ...,\n","                     [-5.4389203e-04, -7.7375676e-04,  7.7704061e-04, ...,\n","                       2.9637743e-04,  7.2399684e-04, -5.1095558e-04],\n","                     [-3.4005992e-04, -2.8000094e-04,  9.0386905e-04, ...,\n","                       1.6101640e-04,  4.6710696e-04,  3.8003526e-04],\n","                     [-5.5585045e-04,  1.4382427e-03, -1.5748385e-04, ...,\n","                       8.8844288e-05, -8.0442245e-05, -8.0268469e-04]], dtype=float32),\n","              array([ 0.00019961, -0.00016256, -0.00055784, ...,  0.00078954,\n","                     -0.00013051,  0.00040907], dtype=float32)),\n","             (array([[-7.3598162e-04, -5.2818691e-04,  1.0908653e-04, ...,\n","                      -3.7775197e-04, -1.7837170e-04,  1.2258939e-04],\n","                     [ 7.2872045e-04,  8.5366197e-04, -5.8743148e-04, ...,\n","                       5.4205034e-04,  1.4244282e-04, -4.5703178e-05],\n","                     [-5.9465977e-04,  7.4781541e-04,  5.2050391e-04, ...,\n","                      -3.8170555e-04,  5.3516572e-04, -1.2766503e-03],\n","                     ...,\n","                     [ 3.4226010e-05,  7.1521901e-04,  1.1743972e-03, ...,\n","                       2.7623473e-04,  2.4821833e-04, -5.7361386e-04],\n","                     [-5.8403128e-04,  3.7044310e-04,  9.2877215e-04, ...,\n","                      -5.1324035e-04,  2.7225021e-04, -4.4859298e-05],\n","                     [-2.7109953e-04,  1.1841779e-03, -7.7749597e-04, ...,\n","                       1.0379138e-03, -4.5211107e-04,  3.7242117e-04]], dtype=float32),\n","              array([-1.7458497e-04, -1.3275449e-04,  2.8735082e-04, ...,\n","                      7.5213575e-05,  1.1741460e-04, -4.8038363e-04], dtype=float32)),\n","             (array([[-4.67696605e-04,  8.23316805e-04, -1.22230442e-04, ...,\n","                       7.73888838e-04,  8.06348165e-04,  3.04043351e-04],\n","                     [-9.91221386e-05, -1.46821607e-04,  7.68509653e-06, ...,\n","                       4.60783020e-04,  2.91097502e-04, -7.07932995e-05],\n","                     [ 3.43673273e-05,  1.13239745e-04, -1.02637801e-04, ...,\n","                      -2.71959307e-05, -3.78697383e-04, -1.46339269e-04],\n","                     ...,\n","                     [ 1.27148756e-04,  2.46073905e-04,  2.98638712e-04, ...,\n","                      -7.13949208e-04, -1.02349638e-03,  3.37633355e-05],\n","                     [-8.59261607e-04,  6.49888185e-04, -7.15430186e-04, ...,\n","                       5.42181719e-04, -7.05597573e-04, -5.23143564e-04],\n","                     [-7.50476975e-05, -8.75051599e-04,  7.74423024e-05, ...,\n","                       2.77646817e-04,  1.58158262e-04, -2.90243421e-04]], dtype=float32),\n","              array([ 3.5663332e-05, -9.7471297e-05, -1.9808193e-05, ...,\n","                      8.1432454e-06,  1.3234805e-04,  3.5878704e-05], dtype=float32))),\n","            (),\n","            (array([[ 4.7496136e-04, -4.4283213e-04, -3.0911100e-04, ...,\n","                     -4.4515546e-04,  2.3411019e-04,  7.0678414e-04],\n","                    [-5.1084481e-04,  8.2031370e-04, -7.6638098e-04, ...,\n","                     -4.7343128e-04, -4.5667149e-04,  8.9457119e-04],\n","                    [ 6.7787495e-04,  2.7011268e-04, -5.5722473e-04, ...,\n","                     -5.3266274e-05,  1.0514038e-03,  5.9659287e-05],\n","                    ...,\n","                    [ 5.0852733e-04, -5.0900795e-04,  5.2043551e-04, ...,\n","                      5.3426373e-04, -1.9353034e-05, -1.3463119e-04],\n","                    [-1.0501333e-03, -2.1320985e-04, -1.3567071e-04, ...,\n","                      6.7303842e-04,  3.0493602e-04,  3.2078088e-04],\n","                    [-7.5355644e-04,  2.4076492e-04,  7.6411967e-04, ...,\n","                     -2.6021659e-04,  4.7376292e-04, -2.6396176e-04]], dtype=float32),\n","             array([ 1.47473445e-04,  1.03243714e-04, -1.63276782e-04, ...,\n","                     1.66554557e-04, -2.20431411e-05, -3.31956806e-04], dtype=float32))), ())),\n","         ()),\n","        ()))),\n","     ()),\n","    (((),\n","      ((), ((array([11.882602, 26.915417, 53.176605, ..., 24.232815, 30.12986 ,\n","                53.490814], dtype=float32),\n","         array([ 0.00037256,  0.00235559, -0.00572654, ..., -0.00085317,\n","                -0.00015001, -0.00319904], dtype=float32)),\n","        (array([[ 9.1061474e-06, -5.2543916e-04,  7.4742158e-05, ...,\n","                  3.4387602e-04, -7.6806196e-04,  9.0095629e-05],\n","                [ 4.4365850e-04, -8.7764522e-04, -2.8442248e-04, ...,\n","                  4.4745041e-04,  2.6851404e-04,  8.2914485e-05],\n","                [ 6.3309667e-04,  6.7067461e-04,  5.6579191e-04, ...,\n","                  3.8447551e-04,  4.2273747e-04,  6.2779710e-04],\n","                ...,\n","                [ 7.4725307e-05, -3.4628752e-05, -9.6878083e-04, ...,\n","                 -9.0135938e-05,  2.2779986e-04,  5.3258456e-04],\n","                [ 7.4696989e-04, -3.3157342e-04, -4.8236220e-04, ...,\n","                 -5.4992776e-04,  1.0120310e-04,  6.4608763e-04],\n","                [ 3.1207327e-04, -1.6867265e-04,  3.8443461e-05, ...,\n","                  5.5873126e-04, -2.0730050e-04, -1.7515484e-04]], dtype=float32),\n","         array([-2.3236156e-04, -3.0561033e-04, -3.0498108e-04, ...,\n","                -2.8665250e-04,  1.6554228e-05, -1.8559807e-04], dtype=float32)),\n","        ((), (), ()),\n","        (),\n","        (array([[-6.42827887e-04,  2.41532165e-04,  2.66172283e-04, ...,\n","                  4.15978488e-04, -2.75289320e-04, -1.08838652e-03],\n","                [-1.78080110e-04, -9.77902440e-04, -4.00595978e-04, ...,\n","                  4.49809275e-04, -7.14282098e-04,  5.74075792e-04],\n","                [-5.71223674e-04,  4.58276074e-04, -5.56489860e-04, ...,\n","                 -5.37563406e-04,  6.17546975e-05, -6.39134123e-06],\n","                ...,\n","                [-1.66098216e-05,  1.82174481e-04,  4.87530750e-04, ...,\n","                  3.90239147e-04,  6.19452840e-05,  2.73486134e-04],\n","                [-6.53528667e-04,  1.46736638e-05, -4.88175283e-04, ...,\n","                 -8.71013108e-05, -3.99038545e-04,  4.84181714e-04],\n","                [-3.75517178e-04,  1.05096886e-04,  3.12700693e-04, ...,\n","                 -5.12888073e-04, -2.47507327e-04, -2.23832263e-04]], dtype=float32),\n","         array([ 9.8310447e-05, -3.6522711e-04,  8.4857328e-04, ...,\n","                 3.7393213e-05,  1.9580405e-04,  6.7399425e-04], dtype=float32)),\n","        ()))),\n","     ()),\n","    (((),\n","      ((),\n","       ((array([23.053377 , 27.30287  ,  0.0849876, ..., 21.66658  , 27.866035 ,\n","                24.262676 ], dtype=float32),\n","         array([ 7.5822099e-05, -3.1709345e-04,  1.3290914e-03, ...,\n","                 1.6320807e-04,  2.9810332e-04,  4.6105927e-04], dtype=float32)),\n","        ((),\n","         ((),\n","          ((), (((array([[ 4.7477541e-04,  1.8920080e-04, -3.5830689e-04, ...,\n","                      -8.2312420e-04,  5.0883432e-04,  1.5199569e-03],\n","                     [ 5.4944266e-04, -1.7542655e-03, -6.9806754e-04, ...,\n","                      -2.1590781e-04,  8.5399218e-04, -1.3554089e-03],\n","                     [ 3.5389391e-04, -1.2240383e-04,  4.1748129e-04, ...,\n","                      -8.8248482e-05, -3.1756409e-04,  7.7933545e-04],\n","                     ...,\n","                     [ 3.0512066e-04,  6.0979096e-04,  7.7297806e-04, ...,\n","                      -7.0093671e-04, -1.2015896e-03, -1.1170958e-03],\n","                     [-6.5673341e-04, -6.5291231e-04, -3.5674855e-04, ...,\n","                      -5.2356090e-06, -7.1201287e-04,  3.8161324e-04],\n","                     [ 4.2226719e-04,  3.9486485e-04,  1.4041309e-03, ...,\n","                       4.6257407e-04, -6.9750834e-04, -1.9814419e-03]], dtype=float32),\n","              array([ 2.6555619e-05,  5.5467754e-05, -4.3178746e-04, ...,\n","                     -1.8974590e-04,  2.2929435e-04, -1.9423856e-04], dtype=float32)),\n","             (array([[ 4.99908754e-04,  3.26675654e-04, -4.49040730e-04, ...,\n","                      -1.12886075e-03,  5.15584718e-04,  1.46994914e-03],\n","                     [ 3.61692146e-05,  3.91253503e-04, -8.63852445e-04, ...,\n","                       5.79612912e-04,  5.51857403e-04, -7.48086837e-04],\n","                     [-1.64948637e-04, -5.57446561e-04, -1.31821865e-03, ...,\n","                       4.24388883e-04,  5.86194568e-04,  3.62798048e-04],\n","                     ...,\n","                     [ 2.87897943e-04, -1.92515843e-04, -3.24249297e-04, ...,\n","                      -6.12052914e-04, -4.04166640e-04, -1.15617251e-04],\n","                     [ 1.01265684e-03, -9.95381197e-05, -1.03045045e-03, ...,\n","                       8.38365697e-04, -1.21160665e-04,  3.50166432e-04],\n","                     [ 8.71289521e-04,  6.75532065e-05,  6.23942411e-04, ...,\n","                       3.69358168e-04, -6.31638744e-04, -1.23738043e-03]], dtype=float32),\n","              array([ 7.6077755e-05, -1.7133288e-04, -2.2819088e-04, ...,\n","                      4.1881456e-05,  3.3688406e-04, -4.0376879e-04], dtype=float32)),\n","             (array([[ 8.8478126e-05,  1.6351017e-04, -8.6873496e-04, ...,\n","                       3.2359894e-05, -2.6871517e-04,  8.1137446e-04],\n","                     [ 6.9618580e-04, -8.5717482e-05, -3.6542802e-05, ...,\n","                       6.5421080e-04,  2.5719349e-04,  3.6893639e-04],\n","                     [ 1.1863918e-04,  1.5579244e-04, -1.6836470e-05, ...,\n","                      -9.1729336e-05, -2.0536114e-04, -1.0182113e-04],\n","                     ...,\n","                     [ 2.9152035e-04,  2.0025258e-05, -1.4103105e-04, ...,\n","                       5.7714054e-04, -8.9075902e-06, -8.8014622e-04],\n","                     [-1.6757916e-04,  1.5152757e-04, -1.5428425e-04, ...,\n","                      -2.6903776e-04, -3.6062591e-04,  6.9527782e-04],\n","                     [-3.3159251e-04, -8.6122578e-05,  4.2896005e-04, ...,\n","                       5.1592413e-04,  4.5576869e-04, -3.4345350e-05]], dtype=float32),\n","              array([-3.6210287e-05,  3.7199574e-05, -9.3787545e-05, ...,\n","                      1.4790458e-05, -2.2550536e-04, -1.9188024e-04], dtype=float32))),\n","            (),\n","            (array([[-3.47191672e-05, -8.96950733e-05,  3.71105241e-04, ...,\n","                     -1.83915603e-04,  2.45887404e-05,  4.51024913e-04],\n","                    [ 2.54542101e-04, -1.32964182e-04,  5.09293284e-04, ...,\n","                      2.49422621e-04,  3.00041458e-04,  6.18076228e-05],\n","                    [ 1.05946348e-03,  3.80291225e-04,  1.04782943e-04, ...,\n","                     -3.30526585e-04, -1.18523625e-04,  5.81236149e-04],\n","                    ...,\n","                    [-4.15631250e-04,  4.41200740e-04,  4.06056846e-04, ...,\n","                      3.00821208e-04, -5.77398496e-05, -2.13037492e-04],\n","                    [-2.13868698e-04,  3.68049776e-04,  5.67135517e-04, ...,\n","                      5.33878629e-05, -8.51450546e-04, -2.79482105e-04],\n","                    [ 2.40833775e-04,  1.48381645e-04, -5.49712968e-05, ...,\n","                      7.80244372e-05,  4.58803639e-04,  9.18613441e-05]], dtype=float32),\n","             array([-7.23109088e-06,  1.08693785e-04, -6.37734542e-04, ...,\n","                     2.54598308e-05,  1.40966495e-05, -3.34856741e-04], dtype=float32))), ())),\n","         ()),\n","        ()))),\n","     ()),\n","    (((),\n","      ((),\n","       ((array([ 6.6400995, 30.494457 , 56.527256 , ..., 25.021791 , 30.356096 ,\n","                49.576252 ], dtype=float32),\n","         array([ 6.0414812e-05,  2.0289321e-03, -1.2403297e-02, ...,\n","                -2.2322659e-03,  1.7856139e-04, -7.3065539e-03], dtype=float32)),\n","        (array([[-4.11537563e-04, -7.19163727e-05, -1.26229192e-04, ...,\n","                  1.16919095e-04,  2.73996877e-04,  4.01998637e-04],\n","                [ 5.79716987e-04,  7.20609212e-04, -7.67348043e-04, ...,\n","                 -2.26146512e-05, -1.36034359e-04, -1.14078785e-03],\n","                [ 1.78121394e-04,  2.89471966e-04,  4.16221679e-04, ...,\n","                  4.37491486e-04,  1.74199813e-04,  2.84798880e-04],\n","                ...,\n","                [-2.11207385e-04, -4.32869012e-04,  6.65298023e-04, ...,\n","                  5.71072102e-04, -4.97211295e-04, -5.75041806e-04],\n","                [ 3.62695020e-04,  3.62479768e-05,  2.78107531e-04, ...,\n","                 -2.37684711e-04, -1.89346785e-04, -1.24536955e-03],\n","                [ 4.81116818e-04,  9.26669687e-04,  5.76918013e-04, ...,\n","                  9.81014105e-04,  1.80177041e-04,  3.32193071e-04]], dtype=float32),\n","         array([-0.00026762, -0.00030415, -0.00039139, ..., -0.00028204,\n","                -0.00027831, -0.00013838], dtype=float32)),\n","        ((), (), ()),\n","        (),\n","        (array([[ 4.5265234e-04,  3.4289624e-04, -1.0637735e-04, ...,\n","                  3.9507099e-04,  4.6197194e-04, -2.9685968e-04],\n","                [-1.8535965e-04, -4.1423229e-04, -9.2393340e-04, ...,\n","                 -6.4259948e-04,  1.0629978e-03, -2.1979737e-04],\n","                [ 3.5849845e-04,  6.2352925e-04,  7.4072467e-04, ...,\n","                  3.1836418e-05,  1.8231988e-06, -1.2871906e-03],\n","                ...,\n","                [-2.3228684e-04, -2.8131955e-04, -7.2644063e-04, ...,\n","                 -1.2809190e-04,  3.3614571e-05,  4.0695080e-04],\n","                [ 2.6384331e-04, -1.2578830e-04, -3.6260465e-04, ...,\n","                  1.2462604e-04, -1.7395485e-06,  1.8278278e-04],\n","                [-4.5465797e-04, -1.1035323e-03, -4.5585146e-04, ...,\n","                 -7.9105783e-04,  1.9356166e-04, -7.1738774e-05]], dtype=float32),\n","         array([ 3.4177287e-05, -2.8308702e-04,  1.0855354e-03, ...,\n","                 1.0730445e-04,  3.9021983e-05,  6.8401155e-04], dtype=float32)),\n","        ()))),\n","     ()),\n","    (((),\n","      ((), ((array([15.46612 , 24.787325,  6.596631, ..., 25.475466, 25.751238,\n","                22.56827 ], dtype=float32),\n","         array([-6.0194725e-05, -1.9631203e-04, -5.6815945e-04, ...,\n","                 2.2254471e-04,  4.0002040e-05,  5.9309579e-04], dtype=float32)),\n","        ((),\n","         ((),\n","          ((), (((array([[-1.1052118e-03, -2.0955016e-03, -2.3717780e-04, ...,\n","                      -7.9861702e-04,  1.0988357e-03, -6.2463764e-04],\n","                     [-1.4289810e-03,  7.8868802e-04,  2.2281933e-04, ...,\n","                      -7.3502096e-04, -3.8259048e-05,  2.8421296e-04],\n","                     [-1.2466379e-03,  2.1735253e-03,  1.1861287e-03, ...,\n","                       1.0206556e-03, -1.4538703e-03, -9.4208098e-04],\n","                     ...,\n","                     [ 3.9622060e-04, -1.8152724e-04, -1.9491832e-05, ...,\n","                       6.2160927e-04,  3.4737699e-05, -4.3466203e-05],\n","                     [-2.5912779e-04, -3.7448981e-04,  7.8589644e-04, ...,\n","                      -1.3088626e-05, -2.8451343e-04,  1.6159791e-03],\n","                     [-1.2939596e-03,  1.1394622e-03,  7.6434476e-04, ...,\n","                      -1.2617020e-03, -1.0391251e-03, -6.7218521e-04]], dtype=float32),\n","              array([ 0.00030503, -0.00095034, -0.00063884, ..., -0.00049736,\n","                      0.00069859,  0.00021916], dtype=float32)),\n","             (array([[-2.1079011e-05, -1.8439755e-03,  1.0862949e-05, ...,\n","                      -1.6025206e-03,  2.1710226e-04, -2.1413530e-04],\n","                     [-1.2431806e-03, -5.3506583e-04,  3.1763790e-04, ...,\n","                       7.9493161e-04, -5.6624407e-04,  7.4198836e-04],\n","                     [-1.0952618e-03,  7.3210377e-04, -1.6938701e-03, ...,\n","                      -7.9208166e-06, -1.3553789e-04,  6.6584337e-04],\n","                     ...,\n","                     [ 3.0796265e-04,  4.3013075e-04,  2.7125908e-04, ...,\n","                       8.9295593e-04,  8.2393992e-04, -2.0104089e-04],\n","                     [-3.5534169e-05, -6.9384690e-04, -7.5421989e-04, ...,\n","                       3.0628746e-04,  3.1676417e-04,  5.1961804e-04],\n","                     [-8.4510772e-04, -6.2127109e-04,  8.0151513e-04, ...,\n","                      -2.7198490e-04, -1.0248568e-03, -1.2529618e-03]], dtype=float32),\n","              array([ 0.0005448 , -0.0007212 ,  0.00010409, ..., -0.00012355,\n","                      0.00024442,  0.00028232], dtype=float32)),\n","             (array([[-2.29504003e-05, -2.96297745e-04, -4.40160569e-04, ...,\n","                       3.25295056e-04, -4.75628476e-04, -3.06185480e-04],\n","                     [ 1.53959001e-04, -4.95536893e-04,  3.55598226e-04, ...,\n","                      -3.53828102e-04, -1.82746109e-04,  3.99724631e-05],\n","                     [ 1.42913850e-04,  4.19956014e-05,  2.35249565e-04, ...,\n","                      -4.28659187e-05, -1.83860553e-04,  9.66568405e-05],\n","                     ...,\n","                     [ 1.57796821e-04, -6.61073602e-04,  1.06728286e-03, ...,\n","                      -2.13956519e-04, -4.72110725e-04, -8.93804943e-04],\n","                     [ 1.15819035e-04,  3.09900381e-04, -4.40212054e-04, ...,\n","                       3.15788959e-04, -7.78638278e-05, -2.94497091e-04],\n","                     [-3.96822958e-04, -2.57213367e-04,  4.79501723e-05, ...,\n","                      -3.99934015e-06, -4.95315180e-04, -4.33730384e-05]], dtype=float32),\n","              array([-2.0235973e-04,  8.4787767e-05,  5.3379381e-06, ...,\n","                     -1.2332921e-05,  6.9274509e-05, -4.6851128e-05], dtype=float32))),\n","            (),\n","            (array([[-8.9315808e-04, -2.6928738e-04, -2.5927782e-04, ...,\n","                     -2.2320799e-04,  1.3464275e-04,  4.6150177e-04],\n","                    [-5.1780441e-04, -5.3752802e-04, -1.5709508e-04, ...,\n","                      1.4199406e-06,  4.0784589e-04,  5.8727543e-04],\n","                    [ 4.3731465e-04,  1.0578498e-03, -2.5544228e-04, ...,\n","                     -3.6957438e-04,  1.8004946e-04, -3.4563051e-04],\n","                    ...,\n","                    [ 2.5125835e-04,  2.0158663e-04,  1.1973290e-05, ...,\n","                     -3.9298018e-05, -2.6337418e-04, -6.6258421e-05],\n","                    [-2.1438002e-04, -6.6393375e-04, -2.5144784e-04, ...,\n","                      7.6797558e-05, -9.0692418e-05,  5.9916556e-04],\n","                    [ 5.5765087e-04,  3.1803738e-04, -5.2228163e-04, ...,\n","                      6.9682224e-05, -2.9040771e-04,  2.5337914e-04]], dtype=float32),\n","             array([-1.5188864e-04, -8.1688275e-05, -6.3122815e-04, ...,\n","                     5.0060662e-05,  7.6495286e-05, -3.0925116e-04], dtype=float32))), ())),\n","         ()),\n","        ()))),\n","     ()),\n","    (((),\n","      ((), ((array([13.304887, 26.722172, 52.879704, ..., 19.950377, 24.642338,\n","                53.150772], dtype=float32),\n","         array([-0.00192272,  0.00293392, -0.00980679, ..., -0.00020195,\n","                 0.00077616, -0.00631617], dtype=float32)),\n","        (array([[ 1.33064852e-04,  2.86546507e-04,  4.20070515e-04, ...,\n","                  5.92719938e-04, -3.73650779e-04,  2.51018122e-04],\n","                [-1.49284300e-04, -3.85249368e-05, -9.62555001e-04, ...,\n","                  3.61030718e-04,  4.65076300e-04,  1.44788786e-03],\n","                [ 4.20487893e-04,  3.04981077e-04,  3.62945953e-04, ...,\n","                  4.26117127e-04,  1.44535108e-04,  6.04928995e-04],\n","                ...,\n","                [ 1.42772205e-05,  9.21776402e-04, -3.89270688e-04, ...,\n","                 -2.76978360e-04,  5.73704951e-04,  7.18284107e-04],\n","                [ 3.11654701e-04,  7.54960347e-04,  2.23587253e-04, ...,\n","                  4.46539052e-04,  3.40628249e-05,  1.25437626e-03],\n","                [-2.44754570e-04, -8.45780087e-05,  2.75057449e-04, ...,\n","                  6.08371629e-04, -4.97398607e-04, -2.87057974e-05]], dtype=float32),\n","         array([-0.00019958, -0.00019813, -0.0002982 , ..., -0.00020765,\n","                -0.00025905, -0.00024065], dtype=float32)),\n","        ((), (), ()),\n","        (),\n","        (array([[-5.21362119e-04, -1.87173151e-04, -1.92125081e-04, ...,\n","                 -1.14817609e-04,  1.94024120e-04, -4.25122154e-04],\n","                [ 1.10604946e-04, -4.69457125e-04,  3.26584122e-04, ...,\n","                  7.84168835e-04, -2.94751924e-04, -3.33678094e-04],\n","                [ 1.59071627e-04,  4.72096028e-04, -2.27829747e-04, ...,\n","                 -2.69612879e-04,  4.07936677e-05,  4.23208403e-04],\n","                ...,\n","                [ 3.39301099e-04,  7.26230384e-04,  2.77083949e-04, ...,\n","                 -3.81413091e-04, -4.87571197e-06,  5.66998555e-04],\n","                [ 5.15278080e-04, -1.57916758e-04,  3.47743517e-05, ...,\n","                 -7.04741222e-04, -1.38067815e-04, -1.68651663e-04],\n","                [ 2.87624367e-04,  7.80637551e-04, -4.00632329e-04, ...,\n","                 -2.49968289e-05,  3.74025607e-04, -3.53332085e-04]], dtype=float32),\n","         array([ 1.00559904e-04, -9.65006548e-05,  1.07303285e-03, ...,\n","                 1.58135546e-04, -2.84488415e-05,  6.27698784e-04], dtype=float32)),\n","        ()))),\n","     ()),\n","    (((),\n","      ((),\n","       ((array([15.900171 , 16.858791 ,  3.2640147, ..., 14.165891 , 16.866034 ,\n","                12.358425 ], dtype=float32),\n","         array([-2.5116769e-04,  3.6935831e-05, -9.5593289e-04, ...,\n","                 1.1980749e-04, -8.4732696e-05,  3.7718751e-04], dtype=float32)),\n","        ((),\n","         ((),\n","          ((), (((array([[-7.6515652e-04, -4.7201256e-04, -2.9925869e-03, ...,\n","                       5.6110154e-04, -1.1543456e-03, -2.6250274e-05],\n","                     [ 7.8548527e-07,  2.3947045e-05, -8.6825277e-04, ...,\n","                      -4.8386530e-04,  7.3704825e-05, -4.1607104e-04],\n","                     [ 1.1968614e-04,  1.8638544e-03,  1.3839048e-03, ...,\n","                      -1.8267155e-03,  1.1112598e-03, -1.8416482e-03],\n","                     ...,\n","                     [ 1.1274805e-03,  1.1643274e-03, -1.6089025e-03, ...,\n","                       6.9829327e-04,  8.3013430e-05, -2.6393702e-04],\n","                     [-9.5984456e-04, -3.9841031e-04,  1.4009990e-04, ...,\n","                       4.5040846e-04,  2.1193807e-04, -1.4974173e-04],\n","                     [ 1.2187173e-03,  3.1799008e-04,  7.7173486e-04, ...,\n","                      -1.1200589e-03,  1.5279624e-04,  2.8405126e-04]], dtype=float32),\n","              array([-0.0002751 , -0.00076521, -0.00045109, ...,  0.00077153,\n","                     -0.00017463,  0.00098568], dtype=float32)),\n","             (array([[-9.19361599e-04,  1.31569558e-03, -7.01303012e-04, ...,\n","                      -3.98322038e-04, -5.58317173e-04,  1.05227833e-03],\n","                     [ 1.18940196e-03,  3.96683550e-04, -3.59350408e-04, ...,\n","                       6.28132257e-04,  7.05603918e-04, -4.05845378e-04],\n","                     [-1.11300440e-03,  5.47096133e-04,  2.40330643e-04, ...,\n","                       1.23881316e-03,  1.37090625e-03,  2.55616542e-05],\n","                     ...,\n","                     [ 1.10974384e-03, -9.29347370e-05,  9.82309459e-04, ...,\n","                       1.20019286e-04,  5.02567913e-04,  2.46788521e-04],\n","                     [-1.38660998e-03, -4.10558481e-04,  8.16197309e-04, ...,\n","                       8.13585721e-05, -9.60538018e-05,  1.05337531e-05],\n","                     [-4.39747237e-04, -4.45615617e-04,  9.66367370e-04, ...,\n","                       1.20207446e-03,  2.60266534e-05, -8.13873557e-05]], dtype=float32),\n","              array([ 5.84244845e-04, -9.83116115e-05,  2.32882812e-04, ...,\n","                      1.05487175e-04, -4.21014229e-05,  5.76377730e-04], dtype=float32)),\n","             (array([[ 1.6677304e-04,  3.8331756e-04,  2.7382126e-04, ...,\n","                      -3.7862919e-05, -1.7537776e-04,  7.6066254e-04],\n","                     [-9.5808494e-04,  5.4590724e-04,  1.0232150e-03, ...,\n","                      -2.0910150e-04, -7.4512576e-04,  7.5189353e-05],\n","                     [ 1.0992933e-05,  1.5086098e-06, -2.4561083e-04, ...,\n","                      -5.2467225e-05,  3.5601723e-05, -1.4986387e-04],\n","                     ...,\n","                     [-2.5354518e-04, -1.1829569e-03,  1.3824442e-03, ...,\n","                       5.6823972e-04, -4.3139365e-04, -3.8781328e-04],\n","                     [-4.0666424e-04, -1.3840213e-06,  1.8210751e-05, ...,\n","                      -6.8994431e-04, -3.7636160e-04,  5.1495525e-05],\n","                     [ 6.6099157e-05, -2.0638788e-04,  1.2980477e-04, ...,\n","                      -4.2876291e-05,  4.7354077e-04,  1.6664057e-04]], dtype=float32),\n","              array([-9.9630546e-05,  9.3340357e-05,  9.1070222e-05, ...,\n","                     -2.1433691e-05,  2.4780255e-05,  6.7533168e-05], dtype=float32))),\n","            (),\n","            (array([[ 5.2150968e-04,  7.3241716e-04, -4.4579431e-04, ...,\n","                      4.7892565e-04,  1.6042193e-04, -5.7003612e-04],\n","                    [ 1.3358182e-03,  6.7485811e-04,  1.9594849e-05, ...,\n","                     -6.4786968e-06,  4.1125523e-04, -7.7357399e-04],\n","                    [-2.8033025e-04, -1.0488561e-03, -6.0570345e-04, ...,\n","                     -4.5426254e-04, -4.8398032e-04, -7.8858144e-04],\n","                    ...,\n","                    [ 5.7595920e-05,  7.3875138e-04, -1.8491077e-05, ...,\n","                     -5.5020595e-05,  3.9005440e-04, -3.3476640e-04],\n","                    [-3.5465448e-04,  5.7606003e-04,  3.9469771e-04, ...,\n","                      3.5845746e-06,  1.6161831e-04,  2.9961867e-04],\n","                    [ 2.3127843e-04,  7.0382339e-05, -5.2502530e-04, ...,\n","                     -3.9035174e-05, -4.7135944e-04,  4.0620347e-04]], dtype=float32),\n","             array([-7.8492827e-05, -7.1773211e-05, -8.5713790e-04, ...,\n","                     3.1634474e-05, -1.6528585e-04, -3.1584510e-04], dtype=float32))), ())),\n","         ()),\n","        ()))),\n","     ()),\n","    (((),\n","      ((), ((array([ 8.447597, 23.9586  , 42.865257, ..., 21.9269  , 27.377716,\n","                45.26886 ], dtype=float32),\n","         array([-1.0831183e-03,  1.3562914e-03, -9.8935151e-03, ...,\n","                -1.5426803e-03,  2.7330125e-06, -5.8466978e-03], dtype=float32)),\n","        (array([[ 8.8930450e-05, -4.9815199e-04,  1.6306138e-04, ...,\n","                  2.3180617e-04, -6.6869246e-04, -5.7863439e-05],\n","                [-3.5536606e-04,  7.1969669e-05,  1.5924354e-04, ...,\n","                 -4.5080518e-04, -1.5396753e-04, -7.9595618e-04],\n","                [ 5.2275751e-05,  2.8476267e-04,  2.7279771e-04, ...,\n","                  3.3925375e-04,  4.6758467e-04,  4.6234406e-04],\n","                ...,\n","                [ 4.4086674e-04,  4.6633664e-04, -2.2051350e-05, ...,\n","                 -6.7695027e-04, -4.5971110e-04, -4.8727819e-04],\n","                [-6.4147403e-04, -3.7969113e-04, -6.0728483e-04, ...,\n","                  7.3826850e-06,  4.5794356e-04,  4.0497931e-04],\n","                [ 5.7575304e-05,  6.6783262e-04,  3.2218255e-04, ...,\n","                  7.2600698e-04, -3.2385662e-05, -7.4884333e-06]], dtype=float32),\n","         array([-0.00010066, -0.00034479, -0.00033323, ..., -0.00035888,\n","                -0.00029662, -0.00018209], dtype=float32)),\n","        ((), (), ()),\n","        (),\n","        (array([[-1.3937031e-04, -1.8485644e-04, -8.9961611e-04, ...,\n","                 -5.8020494e-04, -5.3527352e-04, -3.9621437e-04],\n","                [-2.5579832e-06,  4.4760527e-04, -3.8075430e-04, ...,\n","                  8.4879121e-04, -1.1033806e-04,  2.6250668e-04],\n","                [ 5.6998013e-04, -2.8580253e-04,  2.6485862e-04, ...,\n","                  8.7299768e-04,  6.9006383e-05,  4.6321237e-04],\n","                ...,\n","                [-4.0610315e-04, -3.7333975e-05,  6.0455761e-05, ...,\n","                 -5.7307369e-04,  3.5877316e-04, -3.5970501e-04],\n","                [-7.0543529e-04,  3.7697543e-04,  2.7591144e-04, ...,\n","                 -5.7929673e-04,  1.5935659e-03,  3.9444256e-04],\n","                [-1.5251477e-04,  2.6126497e-04, -1.2432782e-04, ...,\n","                  8.0000621e-04, -6.1706000e-05,  5.7241489e-04]], dtype=float32),\n","         array([ 5.5512832e-05, -2.8135531e-04,  9.3169388e-04, ...,\n","                 1.8353225e-04, -2.1396634e-04,  5.4311415e-04], dtype=float32)),\n","        ()))),\n","     ()),\n","    (((),\n","      ((), ((array([10.323445, 22.997202,  7.536385, ..., 23.583593, 24.953495,\n","                20.02151 ], dtype=float32),\n","         array([-2.40896654e-04, -1.05244944e-04, -1.32396026e-03, ...,\n","                 1.42907942e-04,  2.78596617e-05,  3.83359642e-04], dtype=float32)),\n","        ((),\n","         ((),\n","          ((), (((array([[ 5.7505950e-04, -9.5074967e-04,  6.0425047e-04, ...,\n","                      -1.3985185e-04, -3.7866825e-04, -3.3830176e-04],\n","                     [ 9.9521328e-04,  6.5702503e-04,  7.1052235e-04, ...,\n","                      -1.2056611e-03,  4.0012505e-04,  2.3143753e-04],\n","                     [ 1.3508023e-03,  6.4866629e-04, -1.4113211e-03, ...,\n","                       1.4754479e-03, -7.2914467e-04, -6.5548957e-04],\n","                     ...,\n","                     [-1.3492908e-03,  1.0777600e-03,  3.2638261e-04, ...,\n","                      -1.6990065e-04, -2.7144718e-04, -2.6825501e-04],\n","                     [-2.4706256e-04,  5.9008965e-04,  8.1053760e-04, ...,\n","                       3.5723590e-04, -8.8420429e-04,  1.5710351e-04],\n","                     [-8.6727773e-04, -1.3347443e-03, -6.7518646e-04, ...,\n","                      -4.3859245e-06, -1.4007761e-03, -3.3260352e-04]], dtype=float32),\n","              array([-0.0003806 , -0.00038972,  0.00042433, ..., -0.00075328,\n","                      0.00051525,  0.00044552], dtype=float32)),\n","             (array([[ 7.8727317e-04, -5.2807177e-04,  8.1837852e-04, ...,\n","                      -2.7220283e-04, -1.2325344e-04, -1.0291165e-04],\n","                     [ 2.4799838e-05,  6.0985034e-04, -1.1158101e-03, ...,\n","                      -1.2341095e-03, -2.7948368e-04,  2.9288733e-05],\n","                     [-8.1731909e-04, -1.7078963e-04,  8.6328597e-04, ...,\n","                      -4.7067230e-04, -2.0998699e-04,  1.8305149e-03],\n","                     ...,\n","                     [ 4.8637573e-04, -1.1614701e-03,  5.9785071e-04, ...,\n","                       1.6264599e-04, -5.1646068e-04,  3.6217232e-04],\n","                     [ 6.0022122e-04, -5.9070846e-04, -6.1218312e-04, ...,\n","                      -3.2273514e-04,  2.0310657e-05,  2.8348874e-04],\n","                     [ 6.5886888e-05, -4.4984449e-04, -7.0492510e-04, ...,\n","                       1.0419007e-04,  7.8600994e-04,  1.2692544e-04]], dtype=float32),\n","              array([-9.2846189e-05, -3.0493413e-04, -2.8014506e-04, ...,\n","                     -6.9188995e-05,  3.3318538e-05, -3.2097913e-04], dtype=float32)),\n","             (array([[ 2.4853896e-05,  2.9900306e-04, -6.8152050e-04, ...,\n","                       3.4465408e-04,  3.8031506e-04,  1.9861208e-04],\n","                     [-2.1966464e-04, -1.4190045e-04,  4.7951008e-04, ...,\n","                      -4.2202478e-04,  1.8713443e-04, -7.0625327e-05],\n","                     [-3.3538448e-04,  2.7239103e-05, -1.6623830e-04, ...,\n","                       6.0512863e-05,  2.1410079e-04, -2.6227787e-04],\n","                     ...,\n","                     [ 3.7456196e-04,  3.1411553e-05,  1.0187171e-03, ...,\n","                       2.8893075e-04,  1.2235959e-03, -6.4206810e-04],\n","                     [ 1.9425676e-04, -6.5211701e-04, -2.0656413e-04, ...,\n","                      -3.4269757e-04,  2.5673545e-04, -7.8255351e-04],\n","                     [ 5.5239757e-04, -2.8277942e-05, -6.8722351e-05, ...,\n","                       3.6615855e-04,  1.0613281e-04,  2.4938953e-04]], dtype=float32),\n","              array([ 9.3992930e-05, -4.7856454e-05,  4.4430228e-05, ...,\n","                     -4.5131774e-06, -2.9396064e-05, -7.8593857e-06], dtype=float32))),\n","            (),\n","            (array([[-7.5883538e-05,  4.5163158e-04,  6.6327557e-05, ...,\n","                     -2.5326048e-04, -2.6334924e-04, -1.1345626e-04],\n","                    [ 7.0377314e-06,  1.7299174e-04,  1.2606518e-04, ...,\n","                      6.7523550e-05,  1.0780151e-04, -2.7593895e-04],\n","                    [ 3.6400300e-04, -3.4604740e-04, -1.3661550e-04, ...,\n","                     -6.5954920e-04,  2.3190629e-04,  5.5246177e-04],\n","                    ...,\n","                    [-6.7608577e-04, -2.4983007e-04,  6.8952626e-04, ...,\n","                      4.0843262e-04, -2.0463021e-04, -4.1147042e-04],\n","                    [-4.8224148e-04, -8.9991496e-05, -7.0420581e-05, ...,\n","                     -5.5563165e-04, -4.9088962e-05,  2.4894164e-06],\n","                    [ 1.5214016e-04, -7.0623304e-05, -8.1457867e-05, ...,\n","                      8.5516396e-04,  5.6330819e-04, -3.9715506e-04]], dtype=float32),\n","             array([ 7.5483353e-05,  4.9838527e-05, -8.9394591e-05, ...,\n","                    -6.8716436e-05, -6.5848879e-05, -6.8888345e-05], dtype=float32))), ())),\n","         ()),\n","        ()))),\n","     ()),\n","    (((),\n","      ((), ((array([ 4.519357, 17.095104, 27.162008, ..., 16.009573, 17.475204,\n","                27.181257], dtype=float32),\n","         array([ 0.00091695,  0.00391769, -0.01067202, ..., -0.00174237,\n","                 0.00170684, -0.00561976], dtype=float32)),\n","        (array([[ 4.0900914e-04, -3.6666257e-04, -1.8441380e-04, ...,\n","                  3.3053136e-04,  2.5975195e-04, -3.3156955e-04],\n","                [-5.8955012e-04,  1.1760230e-03,  5.8680528e-04, ...,\n","                 -6.7125529e-04, -9.7770040e-05, -6.7527773e-04],\n","                [ 6.2187208e-04,  1.7220863e-04,  4.1234918e-04, ...,\n","                  4.6112755e-04,  6.4485014e-04,  6.1262108e-04],\n","                ...,\n","                [-1.4884050e-04,  5.3411908e-04, -3.5919854e-04, ...,\n","                  5.7213132e-05,  7.9950485e-05, -3.6008394e-05],\n","                [ 2.0826590e-05, -5.4566492e-04,  1.3221927e-04, ...,\n","                 -2.0174131e-04, -3.2501941e-04, -4.2414994e-04],\n","                [-4.0571176e-04, -3.4323445e-04, -1.0146640e-03, ...,\n","                 -1.5432694e-05, -3.5161016e-04,  2.9592126e-04]], dtype=float32),\n","         array([-0.00046033, -0.00025086, -0.00036944, ..., -0.00043245,\n","                -0.00030267, -0.00040064], dtype=float32)),\n","        ((), (), ()),\n","        (),\n","        (array([[ 1.1463515e-04,  3.4355800e-04,  1.3608503e-04, ...,\n","                  5.3937593e-04,  9.1919588e-05,  3.1221475e-04],\n","                [ 3.4843897e-04, -4.3335187e-04,  6.2250096e-04, ...,\n","                 -1.2909087e-03, -2.9377910e-04, -6.2805717e-04],\n","                [-3.0675842e-04,  2.2982436e-04,  1.7861690e-04, ...,\n","                  2.5261787e-04, -5.8589451e-04, -5.2864873e-04],\n","                ...,\n","                [-1.0410213e-04,  2.3659121e-04, -4.7433216e-04, ...,\n","                 -4.7167053e-04,  9.7032709e-05,  2.4875344e-04],\n","                [-3.0488996e-05,  6.3671329e-04, -2.4590794e-05, ...,\n","                  3.8281144e-04,  4.1998411e-04,  6.1959377e-04],\n","                [ 6.8687441e-05, -7.5059383e-05, -1.1818980e-04, ...,\n","                 -3.2374077e-04,  7.2877257e-05, -1.6137042e-04]], dtype=float32),\n","         array([ 1.3924309e-04, -1.3944639e-04,  1.1164572e-03, ...,\n","                 3.1586154e-05, -1.4708111e-04,  3.5066431e-04], dtype=float32)),\n","        ()))),\n","     ()),\n","    (((),\n","      ((), ((array([13.662609  , 16.912767  , -0.17985429, ..., 19.94888   ,\n","                19.74179   , 11.054346  ], dtype=float32),\n","         array([-2.4868497e-05, -1.5664443e-04, -4.7431805e-04, ...,\n","                 7.0829708e-05,  1.7577445e-04,  3.6049524e-04], dtype=float32)),\n","        ((),\n","         ((),\n","          ((), (((array([[ 1.7954710e-04,  6.3053239e-04, -6.4577471e-04, ...,\n","                      -7.8524890e-06, -5.1341054e-04, -3.6967386e-04],\n","                     [ 3.8966344e-04, -3.3760717e-04,  1.8906455e-04, ...,\n","                      -2.9366658e-04, -1.2506987e-04, -1.1130677e-03],\n","                     [-5.2985555e-04,  9.9893729e-04, -6.7690894e-04, ...,\n","                      -2.3268767e-04, -9.8695408e-04,  1.5939969e-04],\n","                     ...,\n","                     [-4.4666225e-04, -2.2031357e-04, -3.0907904e-04, ...,\n","                      -3.5467366e-04, -3.8284605e-04, -3.3966632e-04],\n","                     [-1.2796897e-04, -5.0379679e-04, -2.8306336e-04, ...,\n","                       4.9571472e-04, -4.8529150e-04, -1.9541119e-05],\n","                     [ 3.9235564e-04,  1.3477897e-03, -5.2949705e-04, ...,\n","                       1.1285846e-03,  5.9983704e-05, -9.4462681e-04]], dtype=float32),\n","              array([ 2.2222640e-04, -3.0761899e-04,  7.6143327e-04, ...,\n","                     -1.0774548e-03,  7.0241437e-04,  5.2318177e-05], dtype=float32)),\n","             (array([[-3.73555609e-04, -1.05851534e-04, -8.35159444e-04, ...,\n","                      -6.73948845e-04,  1.12750533e-03, -1.07906898e-03],\n","                     [ 5.09113052e-05, -5.26391086e-04,  1.07804090e-05, ...,\n","                       9.97368712e-04, -1.05972096e-04, -8.71495940e-05],\n","                     [ 1.60488533e-04, -9.61366895e-05,  8.80712323e-05, ...,\n","                       7.30703410e-04,  1.00479330e-04,  6.70318143e-04],\n","                     ...,\n","                     [-6.18898775e-04, -7.80038681e-05, -1.02008700e-04, ...,\n","                      -5.86262497e-04,  4.69725637e-04, -1.90495106e-04],\n","                     [ 5.04666532e-04, -1.86960126e-04,  1.19793200e-04, ...,\n","                       1.70624626e-04, -1.07888808e-03, -1.06364387e-04],\n","                     [-1.31779734e-05,  1.99126571e-05,  2.50068377e-04, ...,\n","                      -3.15502839e-04,  6.59653742e-04,  1.27806989e-05]], dtype=float32),\n","              array([-0.00012181,  0.00024075,  0.00011933, ...,  0.0001857 ,\n","                      0.00028031,  0.00019143], dtype=float32)),\n","             (array([[ 1.45829108e-04, -1.69275649e-04, -2.74457794e-04, ...,\n","                      -8.81036322e-05,  3.35923047e-04,  1.06226944e-04],\n","                     [-3.91415029e-04, -2.79131582e-05, -9.21985542e-04, ...,\n","                      -4.32961933e-05, -4.46377933e-04, -2.06339653e-04],\n","                     [ 1.32216723e-04,  1.03505707e-04, -1.03300263e-04, ...,\n","                      -5.51262165e-05,  3.76703865e-05,  2.73636659e-04],\n","                     ...,\n","                     [ 8.25936324e-04, -2.18356199e-05, -6.41557155e-04, ...,\n","                       3.99492092e-05,  2.74276012e-04, -7.33335677e-04],\n","                     [ 8.63273570e-04, -4.36983333e-04,  1.43823621e-04, ...,\n","                      -5.28976670e-04, -1.23449703e-04,  8.28841294e-04],\n","                     [-8.18684639e-05,  2.72290024e-04,  2.33570492e-04, ...,\n","                       5.48197131e-04,  3.91247479e-04,  1.36873394e-04]], dtype=float32),\n","              array([-1.9931997e-07, -5.2888248e-05, -2.5585814e-05, ...,\n","                      1.7075001e-04, -2.7265871e-05, -5.0318472e-06], dtype=float32))),\n","            (),\n","            (array([[ 6.5044971e-04,  8.4162661e-05, -6.9227925e-04, ...,\n","                     -4.3800520e-04, -4.2243130e-04, -8.2284212e-04],\n","                    [ 4.7663350e-05,  2.4024997e-04, -5.4797000e-04, ...,\n","                      2.4135892e-04,  6.5453083e-04, -6.8378955e-05],\n","                    [ 1.4151871e-04,  4.1461768e-04, -2.0681704e-04, ...,\n","                      5.4499682e-04,  2.7093102e-04, -5.2287721e-04],\n","                    ...,\n","                    [-3.6583943e-04, -5.1352708e-04,  6.8235234e-04, ...,\n","                      2.4865349e-04,  1.5171844e-04, -8.1020233e-04],\n","                    [ 5.2532437e-04,  4.2408935e-04, -8.6097491e-05, ...,\n","                      1.1302410e-04, -1.5147431e-04, -6.4876216e-04],\n","                    [-6.9541838e-06, -8.5844316e-05, -8.5838401e-04, ...,\n","                      5.7528651e-04, -6.2857888e-04, -4.9598257e-05]], dtype=float32),\n","             array([ 2.6819829e-04, -3.5977784e-05, -9.3013485e-04, ...,\n","                    -9.3305434e-06, -8.0770660e-05,  1.1359405e-04], dtype=float32))), ())),\n","         ()),\n","        ()))),\n","     ()),\n","    (((),\n","      ((), ((array([1.0565677, 1.1246645, 1.3387271, ..., 1.4414002, 1.1400564,\n","                1.6768129], dtype=float32),\n","         array([ 0.00011941,  0.00209474, -0.0075276 , ..., -0.00031828,\n","                 0.00304557, -0.00342649], dtype=float32)),\n","        (array([[ 2.2715289e-04, -7.4326294e-05, -1.7642653e-04, ...,\n","                 -6.9440628e-04, -2.2502949e-04,  2.7442322e-04],\n","                [ 1.3520547e-04,  3.6449201e-04, -2.0915747e-04, ...,\n","                  5.4998149e-04, -5.1587104e-04,  3.7130303e-04],\n","                [ 4.9958366e-04,  3.2166485e-04,  2.4472133e-04, ...,\n","                  1.5018869e-04, -1.3741532e-04,  7.2614773e-04],\n","                ...,\n","                [-1.6568322e-04, -2.0460962e-04, -8.0723170e-04, ...,\n","                  3.3870098e-04,  8.8442815e-05,  1.0088756e-04],\n","                [ 6.1057659e-04,  2.8036826e-04,  5.2802608e-04, ...,\n","                  8.8142348e-04, -3.4302016e-04, -1.4159079e-04],\n","                [ 1.1317254e-03,  3.9756380e-04, -1.1654977e-04, ...,\n","                 -6.6888990e-04,  6.9555477e-04, -1.9448278e-04]], dtype=float32),\n","         array([-0.00123059, -0.00025875, -0.00047418, ..., -0.00022809,\n","                -0.00025056, -0.00027712], dtype=float32)),\n","        ((), (), ()),\n","        (),\n","        (array([[-1.4333241e-03,  9.9568069e-04, -3.7562998e-04, ...,\n","                 -7.4082760e-05, -9.3837189e-05, -1.2070754e-03],\n","                [-3.2182122e-04, -4.6888468e-04,  1.3704115e-03, ...,\n","                 -4.0202736e-04, -6.5621739e-04, -1.6049495e-04],\n","                [-3.2860081e-04, -2.8173352e-04,  2.4195320e-03, ...,\n","                 -1.7784332e-04,  5.7452463e-04, -1.0553803e-04],\n","                ...,\n","                [ 7.1401626e-04, -2.5040319e-04,  4.2948543e-04, ...,\n","                  1.2650220e-03, -9.0488342e-05, -6.8065833e-04],\n","                [ 5.5937510e-04, -8.2099534e-05,  2.3457754e-04, ...,\n","                  2.2208078e-04, -2.4465841e-04, -1.0344024e-03],\n","                [-6.6241901e-04, -1.5787671e-04,  4.5588156e-04, ...,\n","                  6.3233910e-04, -1.7097036e-04, -1.3563718e-04]], dtype=float32),\n","         array([ 9.5904988e-06, -4.9503236e-05, -8.7517037e-05, ...,\n","                -2.8512193e-05, -4.0059469e-05,  7.7080331e-04], dtype=float32)),\n","        ()))),\n","     ()),\n","    (array([15.09562   , 30.442236  ,  0.04095336, ..., 37.74294   ,\n","            37.831093  , 24.789616  ], dtype=float32),\n","     array([ 1.4280340e-04, -5.1927173e-06,  9.8087522e-04, ...,\n","            -6.6376101e-06, -1.6241120e-04, -5.5462396e-04], dtype=float32))),),\n","  (),\n","  ((), (), ()),\n","  {'__marker_for_cached_weights_': ()},\n","  (),\n","  array([[[ 1.05196815e-02,  3.43509857e-03, -1.80868618e-03, ...,\n","            6.77914405e-03,  4.67252638e-03, -5.48521115e-04],\n","          [ 1.27305193e-02, -2.17554928e-03,  6.17686054e-03, ...,\n","            4.19472111e-03,  3.28494422e-03,  5.20140957e-03],\n","          [-2.40533845e-03, -5.72290411e-03, -1.40863940e-05, ...,\n","            7.31986307e-04,  1.64835328e-05,  9.57649667e-04],\n","          ...,\n","          [ 8.89385177e-04, -4.97742416e-03, -4.42023249e-03, ...,\n","            4.94304346e-03,  1.04518258e-03,  4.94702347e-03],\n","          [-3.70781170e-03, -3.43770790e-03, -4.49605286e-03, ...,\n","            4.94293310e-03,  1.04568666e-03,  4.94691916e-03],\n","          [-4.89605218e-03,  1.26261439e-03, -5.72351040e-04, ...,\n","            4.94282227e-03,  1.04619004e-03,  4.94680973e-03]]],\n","        dtype=float32),\n","  ((), ((), ())),\n","  (((),\n","    ((),\n","     ((array([0.59686035, 1.1549416 , 0.8165997 , ..., 1.2106738 , 1.0384041 ,\n","              0.63272065], dtype=float32),\n","       array([-0.00091932, -0.00244375, -0.00254186, ...,  0.00432667,\n","              -0.00166018,  0.0024856 ], dtype=float32)),\n","      ((),\n","       ((),\n","        (((), (((array([[ 1.1609723e-03, -1.0198812e-03,  1.3413309e-03, ...,\n","                      4.0473614e-04, -1.1260419e-03,  6.2519503e-05],\n","                    [-9.5387438e-04, -7.0644601e-04,  5.8748131e-04, ...,\n","                      5.3714821e-04, -1.6110209e-03, -2.5115174e-03],\n","                    [-3.8179688e-04,  5.0085929e-04,  1.2811042e-04, ...,\n","                      7.7093573e-04, -1.2510013e-03, -7.7680475e-04],\n","                    ...,\n","                    [ 1.2662744e-03,  1.4898144e-03, -1.2009869e-03, ...,\n","                     -7.4299477e-04,  9.0430473e-04,  1.7219469e-03],\n","                    [ 1.2213209e-03, -2.0188354e-03,  1.6557080e-03, ...,\n","                     -9.1460842e-04, -1.0496565e-03, -8.1908703e-04],\n","                    [-4.4814133e-04,  2.5739698e-04, -5.1836262e-04, ...,\n","                     -2.3856099e-05,  8.8410144e-04,  4.3200474e-04]], dtype=float32),\n","             array([ 0.00029687,  0.00062775, -0.00045978, ..., -0.00132046,\n","                     0.00155105,  0.00312963], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[ 0.00035386,  0.00528199,  0.00036151, ..., -0.00060325,\n","                     -0.00076432, -0.0084248 ],\n","                    [-0.00832163,  0.00274103, -0.00691348, ...,  0.00223364,\n","                      0.00848608,  0.00254645],\n","                    [ 0.00273388,  0.00172945, -0.00352961, ...,  0.00055839,\n","                      0.00476523,  0.00574508],\n","                    ...,\n","                    [ 0.00370075,  0.00402413, -0.00130196, ..., -0.00588855,\n","                      0.00464505,  0.00342355],\n","                    [-0.00128459,  0.00430714, -0.00444127, ..., -0.00117869,\n","                      0.01034939,  0.00741409],\n","                    [-0.00129762, -0.00277227, -0.00088151, ...,  0.00407334,\n","                     -0.00115117, -0.00159958]], dtype=float32),\n","             array([-4.1926262e-04,  4.6698851e-04, -7.5265672e-04, ...,\n","                     2.9951148e-04, -3.3019244e-04, -7.2159273e-05], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[-2.40492620e-04, -3.48291593e-04,  8.60657237e-05, ...,\n","                      1.65369020e-05,  1.67582562e-04, -4.04661769e-05],\n","                    [ 2.06806391e-04,  1.17358777e-05, -5.27559954e-04, ...,\n","                     -1.01017929e-03,  1.92176420e-04,  2.11350452e-05],\n","                    [ 2.05438249e-04,  1.54929250e-04, -2.26465345e-04, ...,\n","                     -2.49353150e-04, -2.60725428e-05, -1.20510500e-04],\n","                    ...,\n","                    [-8.89141054e-04, -1.05393614e-04,  7.13682850e-04, ...,\n","                      9.98796313e-04, -1.04225757e-04, -7.97538378e-04],\n","                    [-2.73754878e-04,  3.69250571e-04, -2.73452642e-05, ...,\n","                     -1.06026174e-03,  1.10835757e-03,  1.77818467e-04],\n","                    [-6.71463262e-04,  2.67845899e-04, -3.94631206e-05, ...,\n","                     -4.67522361e-04,  4.63829085e-04, -3.51875642e-05]], dtype=float32),\n","             array([ 3.5845111e-05, -2.3148218e-04,  1.1674334e-04, ...,\n","                    -5.1964398e-06,  5.5254418e-08, -1.8359937e-05], dtype=float32)),\n","            ((), (), ())))),\n","         (),\n","         ((), (), ()),\n","         (array([[-4.5426999e-04,  3.6145520e-04,  3.9764817e-04, ...,\n","                  -1.2435652e-04,  8.3506817e-04,  4.0607029e-04],\n","                 [ 4.5603243e-04,  4.2918031e-04, -3.3123241e-04, ...,\n","                  -4.1535654e-04,  7.9064665e-04, -1.9235426e-04],\n","                 [-1.2462259e-04, -2.3822543e-04, -4.7327991e-04, ...,\n","                   2.1172391e-04, -6.5918820e-04,  2.3216907e-04],\n","                 ...,\n","                 [-7.6807395e-05,  1.8043822e-04,  7.6811417e-04, ...,\n","                  -5.5787468e-04,  2.2419399e-04,  2.2080970e-04],\n","                 [-1.0132270e-04, -6.1777263e-04, -2.9545565e-04, ...,\n","                   1.8556051e-04, -4.6798021e-05, -1.6614767e-05],\n","                 [ 3.8658149e-04, -9.3690811e-05, -5.8559753e-04, ...,\n","                   4.1299805e-04, -3.2857360e-04, -1.8212342e-05]], dtype=float32),\n","          array([-4.47413622e-04,  3.43262436e-05,  1.21450124e-04, ...,\n","                  1.23330741e-04, -9.83286882e-05, -3.57135541e-05], dtype=float32))),\n","        ()),\n","       ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([7.9775167, 6.9173894, 7.6066484, ..., 7.018221 , 8.03088  ,\n","              5.46001  ], dtype=float32),\n","       array([-0.00094557, -0.00057708, -0.00065049, ...,  0.0010403 ,\n","              -0.00048297,  0.00036731], dtype=float32)),\n","      (),\n","      ((), (((array([[ 3.3201926e-04, -6.1278325e-04,  1.3920274e-05, ...,\n","                  -4.0700682e-04,  2.8693397e-04, -6.1168976e-04],\n","                 [ 5.9930544e-04,  2.8239407e-05, -6.5187848e-04, ...,\n","                   8.2158404e-05,  5.9858756e-04, -5.8840320e-04],\n","                 [ 1.3858003e-03,  1.3172292e-04,  7.0046185e-04, ...,\n","                  -1.4791483e-04,  7.6851487e-04, -6.4734241e-04],\n","                 ...,\n","                 [-1.0081735e-03,  3.2529960e-04,  6.0811307e-04, ...,\n","                   8.0788566e-04,  1.3124340e-03,  4.9608323e-04],\n","                 [-1.6845675e-04, -3.6319360e-04, -9.0254203e-04, ...,\n","                  -1.4068188e-04,  6.3517143e-04, -5.5983191e-04],\n","                 [ 4.9822766e-04,  2.2683819e-04, -4.3807671e-04, ...,\n","                   4.1688044e-04, -1.2728433e-05,  3.2936421e-04]], dtype=float32),\n","          array([-8.0972683e-04,  8.1698084e-04, -5.2651856e-05, ...,\n","                  8.7798765e-04,  3.2151237e-04,  1.2716308e-03], dtype=float32)),\n","         ((array([[-1.3351710e-04, -1.5392745e-03, -2.6948377e-04, ...,\n","                   -3.0353793e-04,  1.1963282e-03, -3.3233687e-04],\n","                  [ 1.0611096e-03,  8.2757889e-04,  1.1546209e-03, ...,\n","                   -7.0764258e-04,  3.4214373e-04, -5.3402566e-04],\n","                  [ 4.3814880e-04,  8.1519928e-04, -1.6014015e-04, ...,\n","                    1.8195307e-04,  7.4365217e-04,  4.0759047e-04],\n","                  ...,\n","                  [-3.2539101e-04,  2.2494466e-04,  7.5697247e-04, ...,\n","                    6.2766434e-05,  1.4668740e-03, -5.1668502e-04],\n","                  [-9.6434820e-04,  6.1840523e-04,  1.1792715e-03, ...,\n","                    8.8669232e-04,  9.8802603e-04,  3.0324567e-04],\n","                  [-2.4001474e-04,  2.1754785e-03, -5.6620443e-04, ...,\n","                    1.5609701e-03,  1.9680067e-04,  1.9278515e-03]], dtype=float32),\n","           array([ 9.3162780e-06, -7.3333533e-05, -2.7274684e-04, ...,\n","                   5.0996707e-05,  2.5034093e-04,  5.1729279e-05], dtype=float32)),),\n","         ((array([[ 4.0315709e-04, -7.1338771e-05, -3.5068464e-05, ...,\n","                   -5.3557003e-04, -3.7619314e-04,  5.5620963e-05],\n","                  [ 3.5666570e-04, -4.8596086e-04,  7.2997808e-04, ...,\n","                   -3.5072421e-04, -2.0060115e-05, -2.1452195e-04],\n","                  [ 9.5915268e-05,  6.2321793e-05,  9.7713826e-05, ...,\n","                    5.5091880e-04, -5.6629178e-05,  2.4508254e-04],\n","                  ...,\n","                  [ 5.5250468e-05,  3.6460249e-04,  3.7898659e-04, ...,\n","                   -3.0611458e-04, -8.1630162e-04, -2.4302574e-04],\n","                  [ 2.8683725e-04,  4.2023833e-04, -5.7191320e-04, ...,\n","                   -1.5226031e-04, -3.1016709e-04,  4.2354522e-04],\n","                  [ 1.0448856e-04, -6.1639157e-06,  1.5247123e-04, ...,\n","                    8.4097243e-05,  8.8157882e-05,  2.2912370e-04]], dtype=float32),\n","           array([ 2.54584913e-04,  2.69752491e-05, -1.07139575e-04, ...,\n","                   9.62221602e-05,  1.62691547e-04, -1.69557796e-04], dtype=float32)),)),\n","        (),\n","        (array([[-3.07756884e-04,  1.44532940e-04,  4.78536269e-04, ...,\n","                 -4.50232001e-05, -3.24492103e-05,  1.22753729e-04],\n","                [ 3.83376755e-04,  2.55423802e-04,  3.59396188e-04, ...,\n","                  5.92060620e-04, -1.18566801e-04, -3.76437296e-04],\n","                [ 4.91289306e-04,  2.53057719e-04,  6.32201380e-04, ...,\n","                 -5.79283747e-04,  3.52890871e-04,  4.52499546e-04],\n","                ...,\n","                [ 1.96572160e-04,  4.58678143e-04,  8.85469955e-04, ...,\n","                 -4.00763747e-05, -1.21199686e-04,  7.38611619e-04],\n","                [-1.19188837e-04,  2.01978924e-04, -2.03542484e-04, ...,\n","                  4.18914977e-04,  4.55903326e-04, -2.03034535e-04],\n","                [-5.19461697e-04,  2.09209058e-04,  3.06540343e-04, ...,\n","                  8.02842900e-04,  4.48426523e-04, -1.37673167e-04]], dtype=float32),\n","         array([-3.6741717e-04,  7.5127205e-06,  1.6494295e-04, ...,\n","                -3.5311416e-05, -1.5567717e-04,  3.6720435e-06], dtype=float32))), ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((),\n","     ((array([0.3730052 , 0.49164584, 0.46934336, ..., 0.635163  , 0.72636116,\n","              0.40467337], dtype=float32),\n","       array([ 1.2098349e-03, -3.2187425e-04,  4.9736182e-04, ...,\n","               1.3198749e-04,  3.0641739e-05, -4.6996039e-04], dtype=float32)),\n","      (array([[-5.9530447e-04,  6.5970227e-05, -5.6616293e-04, ...,\n","               -9.6208933e-05,  7.0256152e-05,  1.2449129e-04],\n","              [-2.4980551e-04, -2.7903219e-04,  6.2866247e-04, ...,\n","               -1.3705455e-04, -4.4440065e-04, -3.0460936e-04],\n","              [-1.4000294e-04,  4.5881379e-06,  2.0276249e-04, ...,\n","                5.5264725e-05,  2.6077413e-04, -1.0255880e-03],\n","              ...,\n","              [ 3.1886692e-04,  1.2598548e-03,  2.3752890e-04, ...,\n","                3.7036379e-04, -8.7573536e-04, -6.5511966e-05],\n","              [-5.1803677e-04,  6.6446111e-05,  2.0364737e-04, ...,\n","               -3.6826910e-04, -4.2109416e-04, -2.3140708e-05],\n","              [-2.9041272e-04, -4.0379778e-04, -6.8346199e-05, ...,\n","               -1.6226139e-04,  6.2265404e-04,  8.5224304e-04]], dtype=float32),\n","       array([8.1556442e-05, 1.2367158e-04, 3.5060591e-05, ..., 3.7643206e-04,\n","              1.1135120e-04, 5.6016419e-05], dtype=float32)),\n","      ((), (), ()),\n","      (),\n","      (array([[-1.4768934e-04, -8.2443665e-05,  2.2017484e-04, ...,\n","                9.9772536e-05, -3.1859439e-04,  1.3781476e-04],\n","              [ 2.7940227e-04, -6.5910671e-04,  6.5424026e-04, ...,\n","               -2.5552104e-04,  1.6606275e-04,  3.1105836e-04],\n","              [ 2.6224623e-04, -4.9720897e-04,  2.3056639e-04, ...,\n","               -1.1649036e-04, -7.5939123e-04, -2.2293879e-04],\n","              ...,\n","              [ 4.3345630e-04,  1.3848246e-04,  1.5442446e-04, ...,\n","                2.8850543e-04,  3.4136316e-04,  1.3986927e-04],\n","              [-2.6631056e-04,  6.8902655e-04, -7.1626204e-08, ...,\n","                8.3373010e-04,  3.3976819e-04, -9.8044053e-05],\n","              [ 2.8582750e-04,  4.9568474e-04,  2.1441373e-04, ...,\n","                2.4027041e-04,  1.1451069e-04, -6.5853674e-04]], dtype=float32),\n","       array([ 8.8690926e-05,  4.2415104e-06,  1.1052072e-04, ...,\n","              -9.9474391e-06,  1.5786015e-04, -5.7605273e-05], dtype=float32)),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([1.7459681, 1.1203121, 1.5034617, ..., 1.136886 , 1.6519737,\n","              1.0401033], dtype=float32),\n","       array([-0.00962327, -0.00421335, -0.0049278 , ...,  0.00768404,\n","              -0.00458985,  0.00270738], dtype=float32)),\n","      ((),\n","       ((),\n","        (((), (((array([[ 1.4792242e-03, -2.6481336e-03, -2.8025506e-03, ...,\n","                      1.8075368e-03, -6.4058159e-04, -1.7501882e-03],\n","                    [ 3.1672393e-03,  2.4842602e-04,  8.1244111e-04, ...,\n","                      1.3780447e-03,  1.0300298e-04, -6.5381249e-04],\n","                    [ 2.1551760e-04, -9.2783454e-04, -1.3593916e-03, ...,\n","                      6.5077294e-04, -8.4022846e-04, -3.3202738e-04],\n","                    ...,\n","                    [-1.7331120e-03,  1.5347454e-03,  5.5836752e-04, ...,\n","                     -1.4560790e-03, -2.1634885e-04,  8.7856402e-04],\n","                    [-3.4436522e-04, -5.7822157e-04, -1.5923933e-03, ...,\n","                     -7.0378286e-05,  1.0732060e-04,  5.6460506e-04],\n","                    [-8.7207870e-04,  6.4533576e-04, -1.6392724e-04, ...,\n","                     -2.2198565e-03,  1.3669172e-03,  1.6754494e-03]], dtype=float32),\n","             array([-0.0168485 ,  0.01541043,  0.00776885, ..., -0.00577732,\n","                     0.0028181 ,  0.00452609], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[-0.00218979,  0.00394749,  0.00139217, ..., -0.00130369,\n","                      0.00254395,  0.00171812],\n","                    [ 0.00111404,  0.00035449,  0.00435399, ...,  0.00076084,\n","                      0.00139729,  0.00029858],\n","                    [-0.00228976,  0.00455562,  0.00223435, ..., -0.0024383 ,\n","                     -0.00011068,  0.00163428],\n","                    ...,\n","                    [-0.0030313 ,  0.00309369,  0.00114504, ...,  0.00015749,\n","                      0.00098558, -0.00031754],\n","                    [-0.00343178,  0.00173923,  0.00254566, ..., -0.00186504,\n","                      0.00092874,  0.00240577],\n","                    [ 0.00106687, -0.00269217, -0.0022432 , ..., -0.00276931,\n","                     -0.00124202,  0.00108739]], dtype=float32),\n","             array([-3.6565171e-04,  2.1538767e-04, -1.8076099e-06, ...,\n","                    -1.7059912e-04,  4.4070068e-04,  2.4195807e-04], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[-1.31283086e-04,  4.60341798e-05,  5.46151714e-05, ...,\n","                     -1.23656966e-04, -1.15326467e-04, -1.31668858e-04],\n","                    [-1.43533829e-03, -1.11483759e-03, -3.26008361e-04, ...,\n","                      1.63816047e-04, -3.72919982e-04, -6.35761244e-04],\n","                    [ 2.77298212e-04,  6.30820359e-05, -7.60409690e-04, ...,\n","                      3.00999702e-04, -2.59946974e-04, -1.16354575e-04],\n","                    ...,\n","                    [-4.62967175e-04, -3.06420203e-04,  6.15231984e-04, ...,\n","                      9.73006827e-05,  1.34383066e-04,  1.54555353e-04],\n","                    [-7.37772847e-04, -6.94714079e-04, -3.32992749e-05, ...,\n","                     -3.70361173e-04,  3.59794270e-04,  3.56618577e-04],\n","                    [ 7.66228361e-04,  5.44168870e-04,  8.53905687e-04, ...,\n","                     -1.46571823e-04, -3.23441462e-04,  1.71695181e-04]], dtype=float32),\n","             array([-2.8781684e-05,  1.7205425e-04, -4.7767007e-05, ...,\n","                    -1.3462233e-04,  1.8974843e-04,  1.7898421e-04], dtype=float32)),\n","            ((), (), ())))),\n","         (),\n","         ((), (), ()),\n","         (array([[-3.5303106e-04,  2.3849629e-04, -1.4946134e-04, ...,\n","                   6.4884219e-04,  3.3846436e-04,  1.4073275e-04],\n","                 [ 2.2357587e-04, -3.6192822e-04,  2.0763013e-04, ...,\n","                   3.6773138e-04,  4.7014366e-04,  3.4945126e-05],\n","                 [-4.8831396e-04,  9.9627092e-04, -5.2098354e-04, ...,\n","                  -9.6408468e-05, -2.0521600e-05, -2.3153242e-04],\n","                 ...,\n","                 [ 1.5233504e-04, -3.4422276e-04, -4.1325830e-04, ...,\n","                   4.6540253e-04, -9.1272974e-05,  1.3405357e-04],\n","                 [-1.6254776e-04, -6.5409033e-05,  4.2257714e-04, ...,\n","                  -2.4712103e-04, -4.1645209e-04,  1.7567891e-05],\n","                 [ 5.4789084e-04,  4.8553236e-04,  1.2894224e-04, ...,\n","                  -5.7746895e-04,  3.2136260e-04, -2.1158963e-05]], dtype=float32),\n","          array([-2.3404825e-04, -2.9524825e-05,  1.0983164e-04, ...,\n","                 -1.5231088e-04, -7.8076446e-05, -6.2589475e-05], dtype=float32))),\n","        ()),\n","       ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([4.4148974, 4.4752703, 2.9630384, ..., 2.8398435, 4.297058 ,\n","              3.6507745], dtype=float32),\n","       array([-1.5441147e-03, -5.5554317e-04,  4.0130226e-06, ...,\n","               7.6395849e-04, -3.4700733e-04,  8.1990106e-04], dtype=float32)),\n","      (),\n","      ((), (((array([[ 3.60880862e-04,  1.27487452e-04, -4.98239999e-04, ...,\n","                   8.23889684e-04,  5.18401503e-04, -2.23789350e-04],\n","                 [ 1.20299155e-04, -3.30866867e-04, -2.85793474e-04, ...,\n","                   1.42886391e-04, -4.40573436e-04,  3.24504304e-04],\n","                 [ 3.99094628e-04, -1.32194211e-04, -3.08125804e-04, ...,\n","                   8.37058644e-04,  8.11019563e-05, -2.38859720e-04],\n","                 ...,\n","                 [ 7.56380832e-05, -3.75911826e-04,  2.09216145e-04, ...,\n","                   6.39923033e-04,  1.06730976e-03, -3.22762819e-04],\n","                 [ 2.07190227e-04,  2.98743369e-04,  3.13041892e-05, ...,\n","                  -7.63384101e-04,  2.59114837e-04,  8.81729240e-04],\n","                 [ 1.42620542e-04,  2.49637687e-05,  4.00982943e-04, ...,\n","                   4.50941807e-05,  7.08432170e-04,  1.69119638e-04]], dtype=float32),\n","          array([-6.2080473e-04, -4.0040362e-05,  5.1644392e-04, ...,\n","                 -5.3037872e-04, -6.2330917e-05,  3.1172764e-04], dtype=float32)),\n","         ((array([[ 7.3351501e-04,  2.1575007e-04, -3.7642327e-04, ...,\n","                    8.4844866e-04,  3.8887429e-04, -2.9627787e-04],\n","                  [-2.8187720e-04,  6.9924165e-04, -2.2436790e-04, ...,\n","                    4.1296732e-04, -6.1446422e-05, -2.7146659e-04],\n","                  [ 8.0158893e-04, -3.6740288e-04, -5.2811019e-04, ...,\n","                   -6.0615444e-04,  6.2028511e-04,  5.5507082e-04],\n","                  ...,\n","                  [ 3.2711785e-05, -2.2181468e-04,  1.8543067e-05, ...,\n","                   -4.8714367e-04, -3.4993180e-04, -3.1538273e-04],\n","                  [-2.7029017e-05,  1.8888780e-04, -3.9443778e-04, ...,\n","                    6.3007919e-04,  9.5342046e-05, -4.7915956e-04],\n","                  [-1.9472929e-03,  2.6528456e-04,  1.9860507e-03, ...,\n","                   -2.3818011e-03, -2.6860833e-04, -2.8030583e-04]], dtype=float32),\n","           array([ 5.1094312e-06, -3.1274738e-04, -7.8993260e-05, ...,\n","                   1.6143128e-04,  2.5817900e-04, -2.2405421e-04], dtype=float32)),),\n","         ((array([[-6.5485081e-05, -6.7194592e-04,  3.6707078e-04, ...,\n","                   -3.3453823e-04, -2.0560925e-04, -1.4758564e-04],\n","                  [-3.0331116e-04,  4.2272764e-04,  1.4385747e-04, ...,\n","                    2.6643748e-04, -4.1359934e-05,  2.3169362e-04],\n","                  [ 2.5296109e-04, -1.3949119e-05,  1.5808930e-04, ...,\n","                    5.4200606e-05, -1.8240311e-04,  7.2249555e-04],\n","                  ...,\n","                  [-3.1296065e-04, -2.7027476e-04, -5.7443406e-04, ...,\n","                   -1.9460259e-04, -4.4670372e-04,  5.6860305e-04],\n","                  [-2.5486809e-04, -7.7167613e-05, -4.6659933e-04, ...,\n","                   -3.2970918e-04, -1.8250590e-04, -5.4326782e-05],\n","                  [ 5.5341399e-05, -5.4420296e-05, -4.0441187e-04, ...,\n","                    1.6890436e-06, -1.9352019e-04, -1.5713411e-04]], dtype=float32),\n","           array([-8.8147390e-05, -6.4026521e-05,  7.4975847e-05, ...,\n","                  -1.3202196e-04,  3.6378337e-05,  4.3218803e-05], dtype=float32)),)),\n","        (),\n","        (array([[-1.78810849e-04,  2.84177979e-04, -2.34907435e-04, ...,\n","                 -8.08489640e-05,  1.99358008e-04, -4.14831447e-04],\n","                [ 2.36235610e-05,  1.37876676e-04, -3.48305970e-04, ...,\n","                  3.61543120e-04, -2.47468452e-05, -3.11046664e-04],\n","                [-2.03535765e-05,  7.50333929e-05,  2.35219675e-04, ...,\n","                 -1.31009132e-04,  2.78639171e-04, -4.22723213e-04],\n","                ...,\n","                [-2.04873912e-04, -1.03874765e-04, -5.34980092e-04, ...,\n","                  9.78265161e-05, -1.51735279e-04, -9.51855254e-05],\n","                [-5.57502615e-04, -6.54906151e-04, -3.10000411e-04, ...,\n","                 -2.96007463e-04,  4.38501651e-04,  3.81605860e-05],\n","                [ 2.42793089e-04,  4.17504489e-05, -4.50739404e-04, ...,\n","                  5.15866734e-04, -2.63317488e-05,  1.87344020e-04]], dtype=float32),\n","         array([-1.6197019e-04,  1.4544696e-04,  2.5029698e-05, ...,\n","                -1.3569172e-04,  8.2637234e-05, -8.9008812e-05], dtype=float32))), ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((),\n","     ((array([0.02004901, 0.31430864, 0.34779698, ..., 0.4332232 , 0.45260692,\n","              0.3270236 ], dtype=float32),\n","       array([ 1.4924284e-03, -5.6349516e-05,  5.1639514e-04, ...,\n","              -1.7282025e-04,  6.4371986e-04, -7.5117120e-04], dtype=float32)),\n","      (array([[ 1.75769354e-04, -1.29986758e-04, -6.33658376e-04, ...,\n","               -3.24725814e-04, -3.33328615e-04,  4.62611293e-04],\n","              [-3.08901726e-05, -8.83472603e-05, -3.46060318e-04, ...,\n","               -8.72142788e-04,  3.31502757e-04, -7.61192525e-04],\n","              [ 4.98690701e-04,  4.15732245e-07, -4.12255002e-04, ...,\n","               -1.06577216e-04, -5.91372140e-04,  5.67396986e-04],\n","              ...,\n","              [ 3.42645188e-04, -3.61780723e-04, -3.08653922e-04, ...,\n","                6.04327943e-04, -7.26489467e-04,  2.11611259e-04],\n","              [ 9.85644219e-05,  5.54475642e-04, -3.12984979e-04, ...,\n","               -4.54179644e-05,  3.45890185e-05, -3.88723274e-04],\n","              [-1.93274274e-04,  5.52386977e-04, -1.21557321e-04, ...,\n","                2.66555784e-04,  2.58043758e-04,  2.22734394e-04]], dtype=float32),\n","       array([ 8.1825034e-05, -3.3911556e-06, -1.2994558e-04, ...,\n","               1.5704907e-06, -4.0714798e-04, -2.6059625e-04], dtype=float32)),\n","      ((), (), ()),\n","      (),\n","      (array([[ 1.4456408e-04, -3.3564077e-04,  2.9675689e-04, ...,\n","                1.2423870e-03, -4.8783355e-04, -5.7294806e-06],\n","              [-8.6040830e-04,  2.7028346e-04, -3.2876537e-04, ...,\n","               -6.2828978e-05, -2.3942813e-04, -7.6384906e-04],\n","              [-7.0541355e-05,  4.9074838e-04, -3.8330269e-05, ...,\n","               -2.5871544e-04,  5.3159235e-04,  1.2855051e-04],\n","              ...,\n","              [ 1.5529005e-05,  3.7483600e-04,  3.3046413e-04, ...,\n","               -1.3319828e-04, -2.0857860e-05, -4.4901602e-04],\n","              [ 1.1242350e-03,  2.0889350e-04,  5.9478090e-05, ...,\n","                4.6051340e-04, -1.4780572e-04,  3.3909956e-04],\n","              [ 5.8040960e-04,  2.6201206e-04,  4.9102097e-04, ...,\n","                3.4345416e-05, -2.6587179e-04, -8.2229171e-04]], dtype=float32),\n","       array([-1.1055768e-04,  7.3825053e-05,  1.6638832e-04, ...,\n","              -3.4442142e-05,  1.0271248e-04, -6.9387053e-05], dtype=float32)),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([2.3184633, 1.5103097, 1.6784391, ..., 1.162938 , 1.6797388,\n","              1.3019164], dtype=float32),\n","       array([-1.5762496e-03, -3.1177024e-05, -4.1352765e-04, ...,\n","               4.2122885e-04, -4.4173819e-05, -5.9029921e-05], dtype=float32)),\n","      ((),\n","       ((),\n","        (((), (((array([[-2.9626098e-03, -1.8441149e-03, -2.5376792e-03, ...,\n","                     -2.6555284e-04,  1.6686384e-04,  1.2273093e-03],\n","                    [ 4.3535844e-04,  9.7324775e-04,  4.3141400e-04, ...,\n","                     -4.5088187e-04,  5.1926106e-04,  2.4815502e-03],\n","                    [-2.7935324e-05, -5.6577713e-04, -1.3931203e-03, ...,\n","                      8.1145758e-04, -7.8835445e-05, -1.0611396e-03],\n","                    ...,\n","                    [-8.7060861e-04,  6.8738249e-05, -9.1145062e-05, ...,\n","                     -1.1740961e-03,  9.1530575e-04,  8.9999964e-04],\n","                    [-8.9565240e-04, -9.3182037e-04, -6.5519416e-04, ...,\n","                      1.7749247e-05, -6.1434397e-04,  2.4393159e-03],\n","                    [ 7.5985800e-04,  5.7175552e-04,  6.3944008e-04, ...,\n","                     -8.0927392e-04,  4.5194235e-04, -5.8412808e-04]], dtype=float32),\n","             array([ 0.00100262,  0.00157865,  0.00132259, ..., -0.0024176 ,\n","                     0.00196416, -0.00289702], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[ 2.0263484e-03,  1.6501538e-03,  1.4582333e-03, ...,\n","                      1.2099813e-03, -1.8111477e-03, -3.9922101e-03],\n","                    [ 1.3336836e-03,  5.4046884e-04,  4.4245785e-04, ...,\n","                      1.1924487e-03, -2.2950349e-03,  1.7619626e-05],\n","                    [ 3.6415104e-03,  4.0919399e-03,  4.0249145e-03, ...,\n","                     -2.1170799e-03,  2.6773892e-03, -1.3718576e-03],\n","                    ...,\n","                    [ 3.3254570e-03,  2.0328476e-03,  3.0721668e-03, ...,\n","                     -1.7233017e-03,  2.9050931e-03, -8.5529441e-04],\n","                    [ 2.0585400e-03,  9.7387121e-04,  1.7712322e-03, ...,\n","                     -6.4262206e-04,  2.7293476e-04,  4.9229589e-04],\n","                    [-2.7814244e-03, -2.3702215e-03, -2.4784068e-03, ...,\n","                     -1.2962351e-03,  1.1615639e-03, -6.4835954e-04]], dtype=float32),\n","             array([-0.00040609, -0.00025065, -0.00034602, ...,  0.00029192,\n","                    -0.00013578, -0.00020685], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[ 3.1799430e-04, -5.6845340e-04,  2.0176965e-04, ...,\n","                      1.6778683e-04, -2.4432573e-04, -3.0755451e-05],\n","                    [-6.2214280e-04, -2.3700132e-04, -1.5298522e-04, ...,\n","                     -2.4855462e-05, -1.2396570e-04,  1.5312669e-04],\n","                    [-2.2733308e-04,  1.4518898e-04,  2.7525579e-05, ...,\n","                     -1.4783330e-04,  5.9782504e-04, -1.8096968e-04],\n","                    ...,\n","                    [ 5.0378731e-04, -2.6292683e-04,  2.1701300e-04, ...,\n","                      1.2392426e-04, -7.1425061e-04,  5.3465989e-04],\n","                    [ 6.2529976e-04, -2.5314227e-04,  2.8031302e-04, ...,\n","                     -8.1876246e-04,  1.3653157e-04, -1.0695077e-03],\n","                    [-7.1323471e-04,  2.4864453e-04, -1.5005661e-04, ...,\n","                      5.8975525e-04, -6.3729612e-04,  3.4128159e-04]], dtype=float32),\n","             array([-2.79519518e-05, -7.28434243e-05,  8.30739591e-05, ...,\n","                    -9.52003538e-05,  4.34918402e-05, -1.07779815e-04], dtype=float32)),\n","            ((), (), ())))),\n","         (),\n","         ((), (), ()),\n","         (array([[-2.75624159e-04,  8.16262444e-04, -3.55269585e-04, ...,\n","                   1.98135211e-04, -8.44843918e-04,  5.63813373e-04],\n","                 [ 2.52587168e-04, -5.45212650e-04,  7.99670990e-04, ...,\n","                  -3.19525425e-04,  6.90480170e-04, -6.68130582e-04],\n","                 [-2.60266592e-04,  4.83239128e-05,  3.94511953e-05, ...,\n","                  -2.29426514e-04, -2.89813270e-05,  1.42972392e-04],\n","                 ...,\n","                 [ 1.30456219e-05,  2.18000729e-04,  1.18661796e-04, ...,\n","                   1.35222290e-04, -5.01303257e-05,  2.98868748e-04],\n","                 [-3.29772010e-04, -5.94377518e-04, -1.69718594e-04, ...,\n","                  -3.44325643e-04,  4.08747466e-04, -2.22892821e-04],\n","                 [ 3.17795406e-04,  7.51994550e-04, -4.59142932e-04, ...,\n","                   1.88685401e-04, -2.91384604e-05,  6.85180887e-04]], dtype=float32),\n","          array([-1.8052335e-04, -3.7437185e-05,  7.9466728e-05, ...,\n","                 -1.1842687e-04, -9.2524089e-05, -8.0389924e-05], dtype=float32))),\n","        ()),\n","       ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([12.684638, 17.514954, 13.511159, ..., 18.375095, 16.820488,\n","              19.177984], dtype=float32),\n","       array([-3.7566252e-04, -1.1306052e-04, -3.9041548e-05, ...,\n","               1.4574629e-04,  5.7236281e-05,  5.0969247e-05], dtype=float32)),\n","      (),\n","      ((), (((array([[-1.7508499e-04,  1.4247035e-04,  6.5249711e-04, ...,\n","                   5.3085142e-04, -1.4910375e-03,  4.9913727e-04],\n","                 [-6.2512065e-04, -4.9591967e-04, -4.0997055e-04, ...,\n","                   2.0454485e-04,  3.5394056e-04,  4.5141191e-04],\n","                 [ 6.4397353e-04,  7.9244497e-04,  1.8705455e-04, ...,\n","                   4.8923754e-04, -1.2587177e-03,  8.9374260e-04],\n","                 ...,\n","                 [-1.2201700e-04, -1.2548125e-04,  2.0273091e-04, ...,\n","                   5.1917625e-04, -6.4740522e-04, -8.8439451e-04],\n","                 [-1.3839488e-04, -5.4217817e-04, -3.8266269e-04, ...,\n","                  -2.2648487e-03, -1.1818003e-03,  1.2257644e-03],\n","                 [-4.1594444e-04, -2.5428273e-04,  1.5813190e-03, ...,\n","                   1.0825762e-03,  3.2658881e-04, -2.9368646e-05]], dtype=float32),\n","          array([ 7.9547253e-06, -1.3748884e-04,  9.0763235e-05, ...,\n","                  1.0078527e-04, -2.2697901e-05, -5.7396770e-04], dtype=float32)),\n","         ((array([[-1.36675648e-04,  2.26094257e-04, -3.02828295e-04, ...,\n","                    4.82049683e-04,  5.53098740e-04, -2.49718316e-04],\n","                  [-3.23640503e-04, -7.43713987e-04, -3.24642111e-04, ...,\n","                    5.82358567e-04,  3.79588513e-04,  2.00039824e-04],\n","                  [-5.61666908e-04,  7.74579181e-04,  1.98669819e-04, ...,\n","                    4.02617879e-04,  2.95798731e-04, -2.96534970e-04],\n","                  ...,\n","                  [-7.64666416e-04, -5.83993678e-04,  8.59016382e-07, ...,\n","                   -7.96290551e-05,  2.60673231e-04,  1.05574793e-04],\n","                  [-1.14527975e-04, -5.58272004e-04, -3.57791403e-04, ...,\n","                   -2.17265726e-04,  9.92451605e-05,  4.01023339e-04],\n","                  [-1.96486817e-05,  4.28597210e-04,  1.15351162e-04, ...,\n","                    2.90558673e-04,  4.45780228e-04, -8.62439687e-04]], dtype=float32),\n","           array([-5.6683125e-06,  2.8126081e-04,  5.0053332e-04, ...,\n","                  -2.0489648e-04,  5.7058613e-05,  1.5494959e-04], dtype=float32)),),\n","         ((array([[-2.20476300e-04, -1.21569195e-04,  1.91832834e-04, ...,\n","                    2.88261130e-04,  1.89973609e-04,  4.53034125e-04],\n","                  [-1.61372722e-04,  5.17670787e-06,  1.78317350e-04, ...,\n","                   -3.77753488e-04, -9.45448264e-05, -7.55699657e-05],\n","                  [-2.54790095e-04,  5.08641941e-04, -1.70141007e-04, ...,\n","                    4.09083266e-04, -4.25476232e-04, -8.53996335e-06],\n","                  ...,\n","                  [ 3.97519325e-04, -4.72782296e-04,  3.48900008e-04, ...,\n","                   -5.38262131e-04, -6.58818346e-04,  5.35160303e-04],\n","                  [-3.14458593e-04, -7.14644120e-05,  8.38290725e-05, ...,\n","                    3.66595224e-04, -5.99724415e-04,  1.28253487e-05],\n","                  [ 7.20464086e-05,  3.49398906e-05, -1.29730950e-04, ...,\n","                   -4.83851181e-05, -2.14438027e-04,  4.37359762e-04]], dtype=float32),\n","           array([-2.55912092e-09,  7.02855759e-05, -3.09133648e-05, ...,\n","                  -1.03300656e-04,  1.49607862e-04,  2.21353457e-06], dtype=float32)),)),\n","        (),\n","        (array([[-2.0433482e-04, -3.8163307e-05, -1.6210944e-04, ...,\n","                  2.3701884e-04,  2.1140184e-05,  4.8370831e-04],\n","                [ 4.1185692e-04,  1.4733901e-04,  2.7610944e-04, ...,\n","                  2.9915443e-04,  4.0826088e-04, -4.5250342e-04],\n","                [-1.1016715e-04, -2.4325270e-04,  4.8496833e-05, ...,\n","                 -3.0200879e-04,  7.5713753e-05,  7.6207286e-04],\n","                ...,\n","                [ 3.4051624e-04,  1.4948918e-04, -2.4966361e-05, ...,\n","                 -1.7860752e-04,  2.0421320e-04,  1.0967227e-05],\n","                [ 4.2441048e-04,  5.0735725e-05,  2.2870601e-04, ...,\n","                 -5.0522177e-04, -9.2008448e-04, -2.3057274e-04],\n","                [ 5.6345429e-04,  2.8602468e-05, -2.1914837e-04, ...,\n","                  6.3515856e-04, -2.3961085e-04,  2.0971947e-04]], dtype=float32),\n","         array([-2.8552749e-04, -5.7122888e-05,  1.1117451e-05, ...,\n","                -1.7160202e-04, -1.5789805e-05,  3.3680215e-05], dtype=float32))), ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((),\n","     ((array([0.05873281, 1.1386384 , 0.81622845, ..., 1.1089197 , 0.81293654,\n","              1.1210396 ], dtype=float32),\n","       array([ 1.0103113e-03,  9.8843127e-04, -3.7746617e-04, ...,\n","              -7.9459901e-04,  2.2242100e-04, -5.7558569e-05], dtype=float32)),\n","      (array([[-3.5653339e-04, -2.9007345e-04,  2.0596675e-04, ...,\n","                5.5684830e-04, -4.4397311e-04,  9.9103863e-04],\n","              [-6.2109448e-04, -6.2191463e-04, -3.8513541e-04, ...,\n","                2.1407979e-04,  7.6953968e-04,  2.1962529e-04],\n","              [-1.6781599e-04, -2.6846002e-04,  1.5619962e-04, ...,\n","               -5.6805986e-04,  7.5509906e-04, -4.5696081e-04],\n","              ...,\n","              [ 5.2533770e-04,  3.0544307e-04, -3.5541416e-05, ...,\n","               -9.5599575e-04,  2.1229932e-04, -9.5552939e-04],\n","              [-1.8438016e-04,  6.4354524e-04, -2.6638186e-04, ...,\n","               -3.0330970e-04,  8.5739316e-08,  4.1825528e-04],\n","              [-8.0654887e-04,  2.1263245e-04, -2.6491861e-04, ...,\n","               -4.4718458e-04, -2.8510758e-04, -2.5971583e-04]], dtype=float32),\n","       array([-1.1291418e-04,  3.4764231e-05, -2.2317779e-04, ...,\n","              -2.4332282e-04, -2.4963802e-04, -2.0679554e-04], dtype=float32)),\n","      ((), (), ()),\n","      (),\n","      (array([[-9.37783079e-06,  6.00109459e-04,  5.20483591e-04, ...,\n","               -5.29582030e-04, -9.77408723e-04, -6.80965662e-04],\n","              [ 2.56860367e-04,  5.93571691e-04,  3.03859037e-04, ...,\n","               -1.07250475e-04, -5.73744997e-04, -1.27498090e-04],\n","              [ 7.36363363e-05, -8.63996902e-05, -3.69575610e-05, ...,\n","               -7.05477200e-04, -1.07091888e-04, -6.60266436e-04],\n","              ...,\n","              [ 3.35620920e-04, -1.37408671e-04, -5.86976181e-04, ...,\n","               -1.38980919e-04,  2.10708793e-04, -8.43141170e-04],\n","              [-8.95899488e-04, -2.44427094e-04, -1.76750662e-04, ...,\n","               -4.67926351e-04,  4.55222063e-04,  7.42956254e-05],\n","              [ 4.98424692e-04, -1.25927458e-04,  4.26511724e-05, ...,\n","               -2.83254398e-04,  1.83485143e-04, -4.24329737e-05]], dtype=float32),\n","       array([-2.83567177e-04, -2.63017064e-05,  3.33994030e-05, ...,\n","              -2.40317357e-04,  6.04833331e-05,  1.06631975e-04], dtype=float32)),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([5.6097903, 4.7447014, 2.363449 , ..., 2.607866 , 2.952958 ,\n","              3.8080559], dtype=float32),\n","       array([-7.7188847e-04, -1.7178265e-04,  2.4609469e-04, ...,\n","               3.9616384e-04, -9.0011366e-05,  2.8630221e-04], dtype=float32)),\n","      ((),\n","       ((),\n","        (((),\n","          (((array([[-1.16714765e-03,  1.17197959e-03,  8.29710159e-04, ...,\n","                      1.39948307e-03, -8.69676995e-04, -1.46979024e-03],\n","                    [ 8.83784669e-04,  1.40671647e-04, -7.71613792e-04, ...,\n","                     -1.02075483e-05, -5.46331401e-04,  4.35792317e-04],\n","                    [ 3.36098252e-04,  2.70933000e-04, -7.13322661e-05, ...,\n","                      3.10093921e-04, -1.04537106e-03, -1.16850126e-04],\n","                    ...,\n","                    [ 2.21030772e-04,  4.83294571e-04, -1.72786837e-04, ...,\n","                      4.40935197e-04,  7.67629186e-04,  1.39331096e-03],\n","                    [-1.28142652e-04, -1.30317814e-04,  7.37243856e-04, ...,\n","                      9.44078784e-04, -6.28449896e-04, -6.90993969e-04],\n","                    [ 2.10993952e-04, -2.58293148e-04, -6.80203957e-05, ...,\n","                      6.31754694e-04, -5.11264487e-04,  5.92946715e-04]], dtype=float32),\n","             array([ 0.00466574,  0.00073912, -0.00291029, ..., -0.00083776,\n","                     0.00012354,  0.00216888], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[ 2.4482107e-04,  1.7481344e-03, -1.4095301e-03, ...,\n","                      2.8538608e-03, -2.1352503e-03,  8.6887600e-04],\n","                    [-3.2090713e-04, -7.2700978e-04,  3.4817294e-04, ...,\n","                     -4.6165803e-04,  8.0024230e-04, -3.3779262e-04],\n","                    [ 4.3479752e-04,  9.6121588e-04,  1.7643624e-04, ...,\n","                      7.9649914e-04,  3.3137482e-04,  1.2587499e-03],\n","                    ...,\n","                    [ 4.5992780e-04, -5.3132314e-04, -3.8329774e-04, ...,\n","                      1.5028700e-03, -1.5351492e-03,  6.8096025e-04],\n","                    [ 1.3260842e-04, -7.5241079e-04,  6.3113909e-04, ...,\n","                     -1.9646800e-04,  1.4356162e-03,  4.0370089e-04],\n","                    [-1.4745333e-04, -1.3903597e-04, -9.7710232e-05, ...,\n","                      2.8844870e-04,  5.2388423e-05, -8.5744337e-04]], dtype=float32),\n","             array([ 6.4623187e-04,  3.7087261e-04,  9.0773014e-05, ...,\n","                    -3.1318763e-05, -1.2342769e-04,  2.9188787e-04], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[ 3.7146281e-04,  4.6470368e-04, -8.2173119e-06, ...,\n","                     -3.6337873e-04, -1.2265622e-04, -1.3598299e-04],\n","                    [ 1.4827948e-04,  8.6227077e-04,  3.0457156e-04, ...,\n","                     -2.6567770e-05, -3.8639814e-04,  4.3746835e-04],\n","                    [ 2.9339035e-06,  4.5218706e-04,  6.0522336e-05, ...,\n","                     -1.8156477e-04, -5.6822452e-04, -3.5895788e-04],\n","                    ...,\n","                    [ 7.2814917e-05, -1.0556119e-04, -5.3568667e-04, ...,\n","                      7.5304310e-04,  6.0526864e-04, -3.5236048e-04],\n","                    [ 5.6566179e-05,  6.5069515e-05,  4.7263522e-05, ...,\n","                      3.9292750e-04,  2.2461521e-04,  1.4903302e-04],\n","                    [-1.6775576e-04,  3.6158286e-05,  2.2274899e-04, ...,\n","                     -2.2047730e-04, -7.8227476e-04,  4.3158283e-04]], dtype=float32),\n","             array([ 9.56931035e-05,  1.48335530e-05, -1.60124560e-04, ...,\n","                     1.90746112e-04,  6.06652975e-05,  1.06991254e-04], dtype=float32)),\n","            ((), (), ())))),\n","         (),\n","         ((), (), ()),\n","         (array([[-2.63978203e-04, -1.85029014e-04,  2.17268127e-04, ...,\n","                   1.23975711e-04, -4.04881393e-05,  2.27136843e-04],\n","                 [-1.98013746e-04, -1.63666336e-05, -8.97138943e-06, ...,\n","                  -2.66079387e-05,  7.32097484e-04,  2.69356260e-05],\n","                 [-5.37676213e-04, -3.41490057e-04,  1.65243284e-04, ...,\n","                   1.68574887e-04, -1.85679921e-04,  1.34542515e-05],\n","                 ...,\n","                 [-1.08663131e-04,  9.01229505e-05, -3.30551527e-04, ...,\n","                   2.04733398e-04, -1.74051209e-04,  1.85987883e-04],\n","                 [-3.65717133e-05,  2.45238683e-04, -2.45162373e-04, ...,\n","                   1.34491900e-04, -5.48006326e-04,  7.36795249e-04],\n","                 [-1.47864353e-04,  2.95106933e-04,  1.05579842e-04, ...,\n","                  -2.30153106e-04,  3.30668641e-04, -4.53343324e-04]], dtype=float32),\n","          array([-1.9701758e-04, -7.4915966e-05,  1.2063143e-04, ...,\n","                 -4.5417222e-05,  7.8418052e-06, -1.3237620e-04], dtype=float32))),\n","        ()),\n","       ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((),\n","     ((array([10.232999 , 12.615819 , 11.40584  , ...,  5.8786793, 11.490862 ,\n","               9.910576 ], dtype=float32),\n","       array([-0.00056638, -0.00052198,  0.0007114 , ...,  0.00029819,\n","              -0.00068657, -0.00030795], dtype=float32)),\n","      (),\n","      ((), (((array([[-1.38067652e-03,  5.85082918e-04, -5.64436137e-04, ...,\n","                   2.87783041e-04, -7.00385892e-04,  4.82341362e-04],\n","                 [-3.35956283e-04,  1.25019625e-03,  1.95270535e-04, ...,\n","                  -4.67892416e-04,  7.83693613e-05,  2.97124236e-04],\n","                 [ 6.82542668e-05, -8.20961839e-04, -5.40116511e-04, ...,\n","                  -5.72720077e-04,  1.11032662e-03, -1.56621216e-04],\n","                 ...,\n","                 [ 2.72227102e-04, -2.66542133e-06,  7.90649792e-04, ...,\n","                   1.07716222e-03, -7.91518600e-04,  6.36659433e-06],\n","                 [ 2.63512746e-04,  2.78610329e-04,  6.46192348e-04, ...,\n","                  -8.72855599e-04,  1.27867865e-03, -2.10655344e-04],\n","                 [-1.03963815e-04,  4.59657196e-04,  3.31136616e-05, ...,\n","                   1.69817032e-03, -7.91634549e-04, -1.07486802e-03]], dtype=float32),\n","          array([ 6.4404914e-04, -4.0985021e-04,  4.2199827e-04, ...,\n","                 -4.2762779e-04,  3.2698456e-04,  8.5740787e-05], dtype=float32)),\n","         ((array([[-4.6941779e-05, -2.8349046e-04,  4.1625527e-04, ...,\n","                    5.4379698e-04,  4.0178414e-04, -6.5352453e-04],\n","                  [-2.0833206e-04, -2.2687056e-04, -4.8736023e-04, ...,\n","                    1.2050264e-04, -5.7105999e-04,  5.5237580e-04],\n","                  [ 3.8455025e-04, -6.9616782e-04,  5.7826057e-04, ...,\n","                   -3.3671697e-04, -3.3429830e-04,  8.1779767e-04],\n","                  ...,\n","                  [ 6.6545588e-04, -2.2133639e-04,  4.2612889e-04, ...,\n","                   -1.2026412e-04,  1.9743550e-04,  6.1733986e-04],\n","                  [-3.6007966e-04,  2.6334793e-04, -1.9212723e-04, ...,\n","                   -3.8418412e-04,  3.9890685e-04,  1.6455186e-04],\n","                  [ 3.6942412e-04, -2.6413132e-04,  1.3752309e-04, ...,\n","                   -1.5736505e-03,  8.8524545e-04,  5.7019549e-04]], dtype=float32),\n","           array([ 8.4433013e-05, -2.1034527e-04,  6.3903637e-05, ...,\n","                   2.7323232e-04, -2.6396645e-04,  6.1437010e-04], dtype=float32)),),\n","         ((array([[-5.02484036e-04,  8.35242026e-05,  1.09695866e-04, ...,\n","                   -7.47546364e-05, -1.10643450e-04, -1.78401388e-04],\n","                  [-5.16078668e-04,  1.98657202e-04,  2.64567556e-04, ...,\n","                    8.60408181e-05,  4.13802802e-04,  3.65299493e-05],\n","                  [-4.41142212e-04, -4.28066232e-05, -1.34354763e-04, ...,\n","                   -2.68283591e-04, -5.29468780e-05,  3.31399380e-04],\n","                  ...,\n","                  [-1.63312972e-04, -2.91189819e-04, -1.53992209e-04, ...,\n","                    1.09021705e-04,  2.21937633e-04,  1.44930062e-04],\n","                  [ 2.39522546e-04,  3.09292751e-04,  3.47297813e-04, ...,\n","                   -2.54066719e-04, -2.95593374e-04, -3.25253262e-04],\n","                  [ 8.41370260e-04, -2.65181938e-04, -2.08940924e-04, ...,\n","                   -4.40764008e-04, -4.41204233e-04,  1.80752468e-04]], dtype=float32),\n","           array([-5.0508756e-05,  8.1725608e-05,  1.0336225e-04, ...,\n","                   4.7012982e-05, -4.2795746e-05,  4.1783724e-06], dtype=float32)),)),\n","        (),\n","        (array([[-5.4916018e-04, -1.6727828e-04, -1.7548358e-04, ...,\n","                 -5.5426590e-05,  1.2782017e-04, -4.1158465e-04],\n","                [-3.2271561e-04,  1.6267227e-04, -4.8647591e-04, ...,\n","                  2.4286356e-04, -2.5153477e-04, -4.5462092e-04],\n","                [-4.5668115e-05,  3.2543301e-04, -2.3887915e-05, ...,\n","                  1.8932551e-05,  1.0371563e-04,  1.7846606e-04],\n","                ...,\n","                [ 5.6867220e-06,  3.4335320e-04,  2.2073288e-04, ...,\n","                 -1.6942322e-04,  1.0520843e-03, -6.1681254e-05],\n","                [-9.9553290e-05,  6.0171232e-04,  8.4110579e-05, ...,\n","                  3.8449300e-05,  3.0145029e-04,  3.6398115e-04],\n","                [ 1.7999526e-04, -9.2248840e-05,  2.6633919e-04, ...,\n","                  2.8101762e-04, -8.4399566e-05,  9.1255461e-05]], dtype=float32),\n","         array([-1.9795273e-04, -7.2284558e-05,  4.8732778e-05, ...,\n","                -9.8105789e-05,  1.2647492e-04, -6.1279688e-06], dtype=float32))), ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([1.43376  , 3.5905864, 3.6892989, ..., 4.345422 , 3.4296162,\n","              5.3439956], dtype=float32),\n","       array([-0.00065121,  0.0034449 , -0.00019929, ..., -0.00287084,\n","              -0.00047471, -0.00104733], dtype=float32)),\n","      (array([[-6.1780367e-05,  8.4545492e-04,  1.9869229e-04, ...,\n","                5.6611112e-04, -1.3726433e-04,  6.2028394e-04],\n","              [ 3.4029802e-04,  7.8326516e-04,  7.0473802e-04, ...,\n","               -1.0210239e-04, -2.7985579e-05, -5.0092862e-05],\n","              [ 4.9826410e-04, -8.0856594e-04, -3.7281375e-04, ...,\n","               -5.6995841e-04, -2.6251245e-04, -5.1464408e-04],\n","              ...,\n","              [-1.0118639e-03,  5.5886590e-04, -9.0639619e-04, ...,\n","                1.1553596e-03,  3.4927278e-05, -1.4981952e-04],\n","              [ 1.8639051e-04, -6.4874452e-04, -2.1520455e-04, ...,\n","               -5.4042216e-04,  5.2518671e-04,  2.5535905e-04],\n","              [-7.3713163e-05,  4.8373683e-04, -5.5025022e-05, ...,\n","               -1.7984638e-04,  1.3144927e-04, -2.0610105e-04]], dtype=float32),\n","       array([-0.00027832, -0.00010763, -0.00024385, ..., -0.00020066,\n","              -0.00018451,  0.00012492], dtype=float32)),\n","      ((), (), ()),\n","      (),\n","      (array([[ 2.0960777e-04, -1.4533865e-04,  2.6369508e-04, ...,\n","                1.9739702e-05, -4.8647673e-04,  2.7438148e-04],\n","              [-4.7254380e-05, -6.3964428e-04,  6.2882504e-04, ...,\n","               -4.2101985e-04,  1.3435667e-04, -5.0810206e-04],\n","              [-1.7495548e-04, -2.4572175e-04, -2.0825361e-04, ...,\n","                1.1290221e-03,  3.7939168e-05, -8.9960755e-05],\n","              ...,\n","              [-2.3774440e-04, -1.8009471e-04, -4.8984290e-04, ...,\n","               -4.2164951e-04,  2.4829028e-04, -1.7254082e-04],\n","              [ 2.4653500e-04,  3.6653870e-04,  2.3148134e-06, ...,\n","               -1.4636513e-04,  4.5766319e-05, -3.9493424e-04],\n","              [ 1.4131481e-04,  2.1673072e-04, -1.7032765e-04, ...,\n","                8.4004401e-05,  1.2898857e-04,  1.1602451e-04]], dtype=float32),\n","       array([-4.2021283e-04,  4.4063188e-05, -5.5737048e-05, ...,\n","              -1.3174649e-04,  8.3925093e-05,  7.9907928e-05], dtype=float32)),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([6.938461 , 4.1230874, 3.5138001, ..., 3.240917 , 3.4465415,\n","              4.758137 ], dtype=float32),\n","       array([-0.0003893 , -0.00028587,  0.00067424, ...,  0.0001807 ,\n","               0.00018574,  0.00017023], dtype=float32)),\n","      ((),\n","       ((),\n","        (((),\n","          (((array([[-9.13308526e-04,  9.68695895e-06, -2.14762753e-04, ...,\n","                      1.43469381e-03,  1.81533524e-03, -1.89991179e-03],\n","                    [ 4.91849903e-04,  1.07461610e-03, -1.64150129e-04, ...,\n","                     -6.85630250e-04, -5.46681586e-05,  1.02618600e-04],\n","                    [-1.06659357e-03, -1.26526644e-03,  4.24437749e-04, ...,\n","                     -7.97461034e-05,  1.44508871e-04, -2.16650278e-05],\n","                    ...,\n","                    [ 7.85830955e-04,  4.95947897e-04, -3.75969947e-04, ...,\n","                     -9.86921732e-05, -2.09287173e-04, -5.35975887e-05],\n","                    [-2.07257326e-04, -6.87866879e-04,  4.64874785e-04, ...,\n","                      6.16654346e-04,  1.41041703e-03, -6.89759501e-04],\n","                    [ 1.75534212e-03,  4.20337426e-04, -4.32679917e-05, ...,\n","                      1.02968064e-04, -1.54628055e-04, -5.52363927e-05]], dtype=float32),\n","             array([ 0.00053506,  0.00041639,  0.00034594, ..., -0.00057765,\n","                    -0.00052326,  0.00055366], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[-1.6192739e-03, -1.1931472e-03,  1.9028813e-03, ...,\n","                     -2.5464701e-03, -3.0509599e-03,  3.9259451e-03],\n","                    [-4.6653734e-04, -7.0024759e-04,  1.8349875e-04, ...,\n","                     -6.4891728e-04, -1.1197347e-03,  1.1520924e-03],\n","                    [ 3.1521704e-05, -8.5580170e-05, -1.1133316e-03, ...,\n","                      4.0029499e-04,  1.1817713e-03, -1.1665972e-03],\n","                    ...,\n","                    [-4.8965554e-05,  1.1550965e-03, -1.2207819e-03, ...,\n","                     -1.6841461e-04,  9.0353697e-04, -5.4099777e-04],\n","                    [-6.5278588e-04, -8.6604545e-05, -7.8785385e-04, ...,\n","                      1.7023546e-04,  9.7719650e-04, -3.2602463e-04],\n","                    [-1.3340878e-03,  2.6567373e-04, -3.0432909e-04, ...,\n","                      3.3308280e-04, -1.8430004e-04,  3.9185982e-04]], dtype=float32),\n","             array([ 6.3416036e-04,  6.9530035e-04, -1.3141487e-04, ...,\n","                     2.4850719e-04, -2.2461686e-06, -7.1612380e-06], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[-3.5383485e-04,  2.5016748e-04, -4.7819820e-04, ...,\n","                      2.7357970e-04,  2.2644216e-04, -1.0388033e-04],\n","                    [-3.4666227e-05,  6.0591694e-05, -3.0221249e-04, ...,\n","                     -3.3601167e-04, -1.7493639e-04,  2.4812011e-04],\n","                    [ 4.5463335e-04, -5.6657719e-04,  5.8839045e-04, ...,\n","                     -1.9348977e-04, -2.3798076e-04,  3.3137118e-04],\n","                    ...,\n","                    [ 9.5130858e-04, -4.7449573e-04, -4.4427480e-04, ...,\n","                      1.8187084e-05, -2.6162915e-04, -3.4250680e-04],\n","                    [ 8.6799584e-05,  7.1805035e-04, -2.1507022e-04, ...,\n","                     -3.3377501e-04, -8.5102183e-06, -1.5719794e-04],\n","                    [-3.6224403e-04,  2.1656984e-05, -2.3068518e-04, ...,\n","                     -8.7748849e-05, -7.0501890e-05, -7.1173403e-05]], dtype=float32),\n","             array([-8.8088236e-06,  2.2354021e-05,  8.8663081e-05, ...,\n","                    -2.5622901e-05,  5.4111988e-05, -3.7991100e-05], dtype=float32)),\n","            ((), (), ())))),\n","         (),\n","         ((), (), ()),\n","         (array([[-2.32218037e-04,  6.27421541e-05, -1.10115652e-04, ...,\n","                  -4.92519575e-05,  1.30351735e-04,  6.87750988e-04],\n","                 [-5.40442532e-04, -5.91137039e-04,  1.18706535e-04, ...,\n","                   1.46865830e-04, -4.00433724e-04,  6.40812359e-05],\n","                 [ 2.31529615e-04,  1.07018814e-05, -6.06490539e-06, ...,\n","                  -1.57875402e-04, -2.52122496e-04,  3.55538541e-05],\n","                 ...,\n","                 [-3.39547260e-05, -1.80763338e-04, -3.63315630e-04, ...,\n","                   7.64655560e-05,  8.59699867e-05,  2.43579547e-04],\n","                 [-6.28262002e-04,  1.43300029e-04, -9.32807379e-05, ...,\n","                  -2.48305121e-04, -2.60938104e-04, -1.03212806e-04],\n","                 [-2.11428909e-04, -6.79906574e-04,  4.22388461e-04, ...,\n","                  -2.02378011e-04,  2.83235364e-04,  5.94705525e-05]], dtype=float32),\n","          array([ 2.4679635e-05, -8.9719078e-06,  8.6451964e-06, ...,\n","                  3.1825020e-05, -6.0378581e-05, -1.7356974e-04], dtype=float32))),\n","        ()),\n","       ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([19.15786 , 20.113432, 25.305092, ..., 34.89836 , 27.525496,\n","              32.75117 ], dtype=float32),\n","       array([ 1.6305325e-04, -9.3860581e-05, -7.8012337e-05, ...,\n","               4.6053488e-04, -4.1869501e-04,  9.4289491e-05], dtype=float32)),\n","      (),\n","      ((), (((array([[-4.8173522e-04, -1.0815589e-04,  4.9731502e-04, ...,\n","                   3.4944274e-04,  1.3376757e-03, -8.3622534e-04],\n","                 [-1.8906592e-03,  2.1916851e-05,  6.2828750e-04, ...,\n","                  -4.4874789e-04, -5.1633827e-04, -2.0003965e-04],\n","                 [ 9.2917198e-04, -5.7426234e-05, -6.8294321e-04, ...,\n","                  -5.2980095e-04, -9.9479558e-04, -1.8062547e-04],\n","                 ...,\n","                 [ 3.5469708e-04, -1.2084093e-03, -6.3304359e-04, ...,\n","                   1.5646858e-04,  5.5868068e-04,  9.6970279e-04],\n","                 [-1.1865868e-03, -1.2621288e-03, -7.0177892e-04, ...,\n","                  -1.2375236e-03,  4.9017649e-04,  9.6034979e-05],\n","                 [-9.2134648e-04,  6.0477694e-05, -2.7932381e-04, ...,\n","                  -1.5907984e-03, -1.3771940e-03, -1.7170485e-03]], dtype=float32),\n","          array([-3.1563119e-04, -2.1390262e-04,  3.4584235e-05, ...,\n","                  3.3671208e-04, -1.8084964e-05,  1.5191682e-05], dtype=float32)),\n","         ((array([[-1.18960172e-03, -6.69208239e-04,  2.06430312e-04, ...,\n","                    5.98483253e-04, -1.17302094e-04, -1.75408018e-03],\n","                  [-3.99989280e-04,  2.12907486e-04,  1.42371480e-03, ...,\n","                    3.02520348e-04,  5.08668018e-05, -6.52721283e-05],\n","                  [ 9.65331201e-05,  2.48917728e-04,  3.34572745e-04, ...,\n","                   -5.21974987e-04,  1.76620306e-04, -9.43001884e-04],\n","                  ...,\n","                  [ 9.81587000e-05, -1.20301871e-03, -1.41413359e-03, ...,\n","                    2.56068248e-04, -9.42258048e-05,  3.19860352e-04],\n","                  [ 6.51113573e-04, -1.86573574e-03, -1.32628344e-03, ...,\n","                    3.49340917e-05,  5.97752296e-05, -8.34243328e-05],\n","                  [ 6.08320290e-04, -1.58884411e-03,  1.77587863e-04, ...,\n","                    6.06269576e-04,  5.68035175e-04, -1.38187944e-03]], dtype=float32),\n","           array([ 2.0439684e-04,  3.7431059e-04, -2.9999237e-05, ...,\n","                   2.4817145e-04,  5.7644682e-04,  1.5723697e-04], dtype=float32)),),\n","         ((array([[-3.1686644e-04,  4.9466093e-04,  2.0347361e-04, ...,\n","                   -2.6273940e-04,  3.0528661e-04, -1.9762137e-04],\n","                  [ 7.7949831e-04, -5.8269296e-05,  4.1424463e-04, ...,\n","                    4.2903933e-04, -7.4894261e-04,  3.0967334e-04],\n","                  [ 5.0806120e-04, -3.6095566e-04, -7.9782476e-05, ...,\n","                   -8.1053084e-05, -1.5768113e-04, -2.2835098e-04],\n","                  ...,\n","                  [-3.2120694e-05,  3.0024780e-04,  4.0012269e-04, ...,\n","                    2.6529559e-04, -9.8634460e-05,  4.6880238e-04],\n","                  [-4.1392024e-04, -4.5258747e-04,  5.2227819e-04, ...,\n","                   -2.7107695e-04,  3.6284330e-04,  1.5461414e-04],\n","                  [-3.8184831e-04, -3.8000278e-04, -1.7521810e-04, ...,\n","                    5.5302138e-04, -4.7038091e-04,  1.5038045e-04]], dtype=float32),\n","           array([ 1.2551014e-04,  1.2316929e-04,  4.2763702e-05, ...,\n","                   2.6084384e-04,  2.2440606e-04, -1.6938472e-04], dtype=float32)),)),\n","        (),\n","        (array([[-1.2590886e-04,  1.7144176e-04,  3.5075664e-05, ...,\n","                 -3.0111053e-04, -1.9107418e-04,  1.4987082e-04],\n","                [-3.0930090e-04,  1.7713850e-04, -2.6011537e-04, ...,\n","                 -3.9918313e-04, -1.7406528e-05,  1.5027322e-04],\n","                [ 2.0240404e-04, -6.6623325e-04, -9.6185438e-05, ...,\n","                  4.3422615e-04,  4.7860254e-04,  3.1935723e-04],\n","                ...,\n","                [-6.3860381e-04,  1.2133035e-04,  1.5986027e-04, ...,\n","                 -1.8214929e-04,  2.3502178e-04, -1.0240933e-04],\n","                [ 9.3879426e-05,  5.9980334e-04, -1.6021995e-04, ...,\n","                 -2.7532293e-04,  2.1975000e-04, -2.1622013e-04],\n","                [ 2.6116008e-04,  2.0464468e-04, -7.2058283e-05, ...,\n","                  4.9511506e-04,  4.3416442e-04, -3.6017239e-04]], dtype=float32),\n","         array([-2.8286994e-04, -1.2987587e-04, -1.3680967e-05, ...,\n","                -1.0966513e-04, -1.1690583e-04,  1.1014721e-05], dtype=float32))), ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([2.2478302, 7.762549 , 7.8002543, ..., 8.469657 , 6.0435953,\n","              8.595118 ], dtype=float32),\n","       array([ 1.6703843e-03, -1.0369017e-04,  3.2087395e-04, ...,\n","              -2.3257979e-03,  7.1293360e-04,  1.4435094e-05], dtype=float32)),\n","      (array([[-2.04658645e-04, -3.98469390e-04,  9.79083925e-05, ...,\n","                4.44270190e-05,  2.08105441e-04,  4.15472139e-04],\n","              [ 1.00375689e-03, -1.11820736e-04, -3.79837671e-04, ...,\n","               -4.66380989e-05, -1.06404164e-04, -1.09680484e-04],\n","              [-4.11435874e-04,  3.26236594e-04,  4.74972563e-04, ...,\n","               -2.57277949e-04,  5.23144729e-04,  1.12391972e-04],\n","              ...,\n","              [-9.63194907e-05,  6.59326732e-04,  1.51070300e-04, ...,\n","                4.26749379e-04,  6.76080992e-04, -3.46726098e-04],\n","              [-4.08796972e-04,  9.67866450e-04,  4.63682169e-04, ...,\n","                9.11876850e-05,  4.07064013e-04,  2.46050098e-04],\n","              [ 8.56895698e-04,  7.36080838e-05,  2.60934932e-04, ...,\n","                3.30498966e-04,  6.89833192e-04, -1.64009689e-05]], dtype=float32),\n","       array([-0.0002812 , -0.00015736, -0.0001891 , ..., -0.00020013,\n","              -0.00014221, -0.00016924], dtype=float32)),\n","      ((), (), ()),\n","      (),\n","      (array([[-5.8798242e-04, -9.4499328e-06,  8.7558472e-04, ...,\n","               -4.6893794e-04, -1.8979919e-04,  5.5893790e-04],\n","              [-1.0689640e-04,  5.3250964e-04,  1.8832172e-04, ...,\n","                7.6487158e-05,  1.0064104e-04,  5.2061927e-04],\n","              [ 6.9172779e-04, -2.3950262e-04,  3.3480156e-04, ...,\n","                3.3340475e-04, -2.4119914e-04, -2.6150522e-04],\n","              ...,\n","              [ 5.1534432e-04,  1.2324628e-04, -1.4617990e-04, ...,\n","                5.5930504e-05, -2.1252864e-04, -9.3360426e-04],\n","              [ 6.9435843e-04, -2.9642429e-04, -6.3573412e-04, ...,\n","               -3.1235837e-04,  4.1015417e-04,  2.3516527e-04],\n","              [-5.6769222e-04, -1.2750266e-04, -8.5194151e-05, ...,\n","                9.7142773e-05, -1.3695657e-04, -8.4076892e-05]], dtype=float32),\n","       array([-5.59589243e-04, -4.70638624e-05, -7.25490900e-05, ...,\n","               1.08491775e-04,  1.83803219e-04,  9.22709587e-05], dtype=float32)),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([17.629148, 15.73509 , 17.130781, ..., 13.901814, 13.242755,\n","              16.295004], dtype=float32),\n","       array([ 1.3465247e-04, -1.9697017e-04,  2.5829332e-04, ...,\n","               6.1508041e-04,  8.0044971e-05,  4.4975552e-04], dtype=float32)),\n","      ((),\n","       ((),\n","        (((), (((array([[-5.5372965e-04,  4.2218633e-04,  3.3471177e-04, ...,\n","                     -2.9729097e-04, -8.5490954e-04, -6.9313229e-04],\n","                    [-3.2560475e-04,  9.1302977e-04, -1.1515367e-03, ...,\n","                      4.2002057e-04, -2.7087878e-04, -7.8841834e-04],\n","                    [-7.2997867e-04,  2.8634249e-04, -9.0996822e-04, ...,\n","                      7.7166209e-05,  5.1910977e-04, -1.7451991e-04],\n","                    ...,\n","                    [-5.7219677e-05,  3.0876213e-04, -4.5472648e-04, ...,\n","                      2.1355826e-04, -9.8644034e-04,  1.3464449e-04],\n","                    [-1.2208645e-03, -1.4792568e-04, -1.1540155e-03, ...,\n","                      4.3125276e-04,  5.8484974e-04, -5.4237788e-04],\n","                    [ 1.7665085e-04, -3.8951253e-05, -2.8615166e-04, ...,\n","                     -6.0306635e-04,  8.1151363e-04,  5.5813725e-05]], dtype=float32),\n","             array([-1.0307021e-04,  3.2633462e-04, -7.9164420e-05, ...,\n","                    -4.6236708e-04,  5.1521547e-05,  3.4660788e-04], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[-1.3230412e-04,  1.1115703e-03, -4.5041594e-04, ...,\n","                     -3.9017576e-04, -1.0293649e-03, -9.4043557e-04],\n","                    [ 9.3416002e-04,  3.6184624e-04, -1.4951511e-03, ...,\n","                     -1.8948462e-04, -6.5026234e-04, -2.1678113e-04],\n","                    [-3.5697650e-04, -2.1070823e-04,  1.6790400e-03, ...,\n","                      4.8013055e-04, -2.6162798e-04,  1.9175703e-04],\n","                    ...,\n","                    [ 6.8947807e-04, -5.7214824e-04, -6.1120122e-04, ...,\n","                      4.6352085e-04,  2.8906274e-04,  4.7758810e-04],\n","                    [-2.1498065e-04,  2.8573305e-04,  5.4155121e-04, ...,\n","                      7.8894000e-04,  1.0038512e-03,  1.5131221e-04],\n","                    [ 4.1777454e-05,  3.9524151e-04,  5.5824267e-04, ...,\n","                     -2.9091039e-04,  1.3286772e-03, -8.5665920e-04]], dtype=float32),\n","             array([-1.3340756e-04,  5.8867689e-04,  5.7826255e-05, ...,\n","                    -1.7444207e-05, -1.7744753e-05, -3.2906854e-04], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[ 1.3506015e-04,  1.7901107e-04, -4.6092231e-05, ...,\n","                     -1.8292219e-04, -6.8461304e-05, -5.0363876e-04],\n","                    [ 7.5571265e-06,  4.2837035e-04, -1.1273841e-04, ...,\n","                     -5.4793234e-04, -2.9419415e-04, -1.0010654e-03],\n","                    [ 4.2668302e-04, -5.9107644e-04, -5.5615429e-04, ...,\n","                     -7.8190042e-06,  5.2936353e-05,  5.1971653e-04],\n","                    ...,\n","                    [ 3.2973723e-04,  4.3568073e-04, -1.3019450e-04, ...,\n","                      4.2092419e-04, -4.8960771e-08, -2.7475497e-04],\n","                    [-8.0470159e-04,  2.2240642e-05,  4.1284124e-04, ...,\n","                     -2.4311655e-04, -3.3644142e-04,  1.6349446e-04],\n","                    [ 6.8401400e-04,  2.8769978e-04, -1.6145239e-04, ...,\n","                      1.6630882e-04,  5.1611377e-04,  7.0988599e-06]], dtype=float32),\n","             array([-8.9737878e-05,  2.7490983e-04,  1.6830754e-04, ...,\n","                    -2.6737456e-04,  2.3715633e-04, -1.7960738e-04], dtype=float32)),\n","            ((), (), ())))),\n","         (),\n","         ((), (), ()),\n","         (array([[-2.1410809e-04,  3.1062792e-04, -5.0093501e-04, ...,\n","                   2.2426702e-04, -2.4614032e-04,  2.7788259e-04],\n","                 [ 2.1905567e-04,  3.5187233e-05, -4.7458531e-04, ...,\n","                  -5.2221178e-05,  6.0089317e-04, -2.5827702e-04],\n","                 [ 1.8551288e-04, -2.3526781e-04,  2.5294200e-04, ...,\n","                  -2.3776980e-04,  1.1285108e-03, -6.0868002e-05],\n","                 ...,\n","                 [-5.5104290e-04,  4.6464830e-04, -4.8321199e-05, ...,\n","                  -1.8407367e-04,  2.1590901e-04,  1.4477839e-04],\n","                 [ 1.0838595e-04,  1.1481004e-04,  2.1600342e-04, ...,\n","                   2.3667325e-04,  3.3570628e-04, -2.7949613e-04],\n","                 [-9.7866206e-05,  6.0851098e-04, -6.8601896e-04, ...,\n","                   3.2950536e-04,  9.2902032e-05,  3.5576447e-04]], dtype=float32),\n","          array([-6.5917833e-05, -7.9400364e-05, -3.9114406e-05, ...,\n","                  6.5377603e-06, -4.8067086e-05,  5.2639875e-06], dtype=float32))),\n","        ()),\n","       ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([15.348426, 15.621089, 12.522452, ...,  7.897455, 11.191346,\n","              10.001051], dtype=float32),\n","       array([-9.0956868e-04, -9.5881885e-05,  1.1248278e-04, ...,\n","               7.4972209e-05, -7.1362476e-04, -6.1378616e-04], dtype=float32)),\n","      (),\n","      ((), (((array([[ 1.4334747e-03,  1.6318103e-05,  2.9945077e-04, ...,\n","                   8.9068621e-05, -1.0954718e-03, -1.3859045e-04],\n","                 [-6.8791868e-04,  8.1920490e-04,  9.9211007e-05, ...,\n","                   1.2699568e-03, -1.1137679e-03,  3.2003899e-04],\n","                 [ 6.0520577e-04, -8.7395958e-05, -1.6747457e-03, ...,\n","                  -2.5000673e-04, -3.4598904e-04, -3.0652943e-04],\n","                 ...,\n","                 [ 3.4076156e-04, -5.0005037e-04,  9.6510956e-04, ...,\n","                   5.9667544e-04, -1.1799606e-03,  9.0589274e-06],\n","                 [-4.1832688e-04,  4.5775875e-04,  1.9592972e-04, ...,\n","                   1.5817033e-04, -4.3010383e-04,  2.8771849e-04],\n","                 [-9.3370814e-05, -5.2641443e-04,  2.1333189e-04, ...,\n","                  -2.7770319e-04,  7.3240401e-04,  3.1756947e-04]], dtype=float32),\n","          array([-6.6551834e-04,  3.5042688e-04, -1.7899918e-04, ...,\n","                 -1.7352328e-04, -9.5239693e-06, -5.0185237e-04], dtype=float32)),\n","         ((array([[ 1.4590943e-03, -4.7242461e-04, -2.1348693e-04, ...,\n","                    7.7417336e-04, -5.4730568e-04,  2.5966787e-04],\n","                  [-3.3669008e-04, -8.7219011e-04,  7.9205853e-04, ...,\n","                    1.7377334e-04,  1.4360186e-04,  4.8422758e-04],\n","                  [-1.4345658e-06, -3.6238975e-04, -3.7039720e-04, ...,\n","                    6.6361798e-04,  4.7355206e-04, -6.4833357e-04],\n","                  ...,\n","                  [ 5.7595727e-05, -1.3751831e-03,  1.5602604e-04, ...,\n","                   -3.5502447e-04, -1.3838687e-04,  1.0802249e-04],\n","                  [ 2.1531712e-04, -1.0350524e-03,  3.3885575e-04, ...,\n","                    5.5528203e-06,  5.6255906e-04,  8.0303702e-04],\n","                  [-1.5318341e-03,  6.7570584e-04, -2.9617062e-04, ...,\n","                    1.2884632e-03,  4.2965263e-04, -8.1887859e-04]], dtype=float32),\n","           array([-0.00029195,  0.00030242, -0.00015688, ...,  0.00019806,\n","                  -0.00033496,  0.00011893], dtype=float32)),),\n","         ((array([[-1.9561470e-04, -2.5075980e-04,  9.5120624e-05, ...,\n","                   -4.7266626e-04, -3.1297430e-04,  5.4975837e-05],\n","                  [ 2.9336219e-04,  4.6529804e-04, -2.6727846e-04, ...,\n","                    1.4706260e-04,  2.8595017e-04, -8.0383563e-04],\n","                  [-1.7807570e-04, -6.7504538e-05,  2.9765624e-05, ...,\n","                    1.0355671e-04, -3.4724560e-04, -4.2743571e-04],\n","                  ...,\n","                  [ 1.7317649e-04, -4.5996247e-05, -3.2185233e-04, ...,\n","                    1.5226119e-04,  3.8631223e-04, -8.1308058e-04],\n","                  [-2.1136037e-04, -1.1282032e-04,  2.3539062e-04, ...,\n","                    6.5650279e-04, -1.0492279e-04,  5.8931415e-04],\n","                  [ 1.6302701e-04, -1.4911262e-04,  3.0586161e-04, ...,\n","                    2.1102423e-04, -6.8067187e-05,  3.5586931e-05]], dtype=float32),\n","           array([ 9.0192429e-05,  6.5161643e-05, -5.0859268e-05, ...,\n","                   2.1839226e-04, -9.8054363e-05,  2.4391279e-04], dtype=float32)),)),\n","        (),\n","        (array([[ 2.5522689e-04, -2.3672901e-04, -1.0058208e-04, ...,\n","                 -4.1208442e-04, -5.2475807e-04,  7.7749777e-04],\n","                [ 3.1925284e-04, -3.6952755e-04, -2.4554387e-04, ...,\n","                 -1.6339697e-04,  1.2273766e-04, -1.9287779e-04],\n","                [-1.9801321e-04,  3.0239766e-05,  3.9562650e-04, ...,\n","                 -2.1933280e-04,  1.8021537e-04, -7.9285051e-04],\n","                ...,\n","                [-4.6249121e-04, -4.8173490e-05, -1.8148481e-04, ...,\n","                 -4.7332788e-04,  5.1376141e-05,  3.2335197e-04],\n","                [ 5.7904101e-05,  2.8426040e-04,  3.6702506e-04, ...,\n","                  2.2211738e-04, -2.4319992e-04,  3.2044267e-05],\n","                [ 4.1102205e-05,  6.2265062e-05, -4.8913836e-04, ...,\n","                 -1.0914252e-03, -5.7217909e-04,  6.2330998e-04]], dtype=float32),\n","         array([-5.1531519e-05, -3.7472360e-05, -1.4277217e-04, ...,\n","                -2.0263924e-06,  6.7443980e-06, -2.1003652e-05], dtype=float32))), ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([ 7.445763, 10.484303, 10.578824, ..., 12.412859,  9.58899 ,\n","              13.310836], dtype=float32),\n","       array([ 1.2864274e-04, -6.1849691e-04,  8.3369363e-05, ...,\n","              -1.0331027e-03, -8.8098814e-04, -2.5927465e-04], dtype=float32)),\n","      (array([[-4.5694894e-04,  8.7851542e-04, -2.2541239e-05, ...,\n","                3.8457674e-04,  1.0744151e-06,  3.6295527e-04],\n","              [ 9.1066532e-04,  9.0452982e-04, -7.2598719e-04, ...,\n","               -3.1369406e-04,  5.2463321e-04, -6.3186111e-05],\n","              [ 6.6953464e-05,  1.2834117e-04, -3.2320016e-04, ...,\n","                9.8891370e-04,  3.8195433e-04,  5.8448385e-04],\n","              ...,\n","              [-1.1883340e-04,  4.5967539e-04,  4.9997889e-04, ...,\n","               -3.8770842e-04,  6.3031061e-05,  1.6457344e-04],\n","              [-1.4330364e-04,  3.6158427e-04,  3.6584770e-05, ...,\n","                3.6519204e-04,  7.3528063e-04, -8.5272000e-04],\n","              [ 3.9997275e-04,  7.0268009e-04,  1.0124858e-03, ...,\n","                1.7136169e-04,  3.4794939e-04, -5.0839223e-04]], dtype=float32),\n","       array([-0.00031749, -0.00029829, -0.00031444, ..., -0.00015337,\n","              -0.00025174, -0.0002712 ], dtype=float32)),\n","      ((), (), ()),\n","      (),\n","      (array([[-5.4707320e-04, -2.6726798e-04, -3.2912625e-04, ...,\n","               -1.4641540e-04, -3.2520152e-04,  4.3039968e-06],\n","              [-1.8418563e-04,  3.5802525e-04, -6.1326238e-05, ...,\n","               -2.8304890e-04, -5.7748264e-05,  6.7675853e-04],\n","              [ 3.4287979e-04, -1.5087973e-04, -2.3377860e-04, ...,\n","                1.7684547e-04, -1.0309740e-04,  2.9595927e-04],\n","              ...,\n","              [-4.2899926e-05, -3.1472187e-04, -3.0540890e-04, ...,\n","               -2.3605737e-04,  7.3279440e-04,  2.6072055e-04],\n","              [ 3.0024870e-05, -3.3559499e-04, -1.9136343e-04, ...,\n","                2.1946820e-04, -2.7276957e-04, -3.8305021e-04],\n","              [ 5.7839212e-04,  4.5549794e-04, -5.1957782e-04, ...,\n","                1.2581679e-04,  1.1238064e-04,  2.0314871e-04]], dtype=float32),\n","       array([-2.6389500e-04,  3.1592746e-04, -7.6882636e-05, ...,\n","               1.8918154e-05,  2.3277619e-04, -2.0722373e-04], dtype=float32)),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([18.0873  , 21.569647, 16.152906, ..., 23.175999, 18.336205,\n","              20.850653], dtype=float32),\n","       array([ 7.8933981e-06, -8.9008590e-06,  4.2394799e-04, ...,\n","               2.5672492e-04,  3.0615513e-04,  5.6972116e-04], dtype=float32)),\n","      ((),\n","       ((),\n","        (((), (((array([[ 3.9929894e-04,  1.5782073e-04, -9.3741619e-05, ...,\n","                     -3.3261528e-04, -1.4817102e-03, -4.4762279e-04],\n","                    [-8.5309439e-04,  1.1111036e-03, -1.3440505e-03, ...,\n","                      3.0209767e-04, -9.0436183e-04, -1.0773800e-03],\n","                    [ 5.6494231e-04, -1.7102031e-04, -6.3793850e-04, ...,\n","                      5.4368679e-04, -4.6966510e-04, -8.3978701e-04],\n","                    ...,\n","                    [ 1.0584414e-03, -1.8332754e-03, -4.3729009e-04, ...,\n","                      1.4548257e-04,  7.2097644e-04,  4.8021376e-04],\n","                    [-3.4601660e-04, -1.1746201e-03, -7.6963872e-05, ...,\n","                     -5.1820895e-04,  8.9489258e-05,  1.1814634e-03],\n","                    [ 1.3998819e-04, -7.2288397e-04, -5.1633315e-04, ...,\n","                     -4.9147947e-04, -5.3388980e-04,  2.2530361e-04]], dtype=float32),\n","             array([ 6.0057041e-04,  8.2051993e-05,  2.2924585e-05, ...,\n","                     1.7890395e-04, -1.4356349e-04,  6.0344453e-05], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[ 3.1815344e-04, -8.7075896e-04, -1.4093236e-04, ...,\n","                      8.2025933e-04,  2.3268652e-04, -1.7704481e-04],\n","                    [ 2.3838165e-04,  1.3752602e-04, -1.6951996e-04, ...,\n","                     -2.5610352e-04,  1.2118677e-03, -2.0930779e-03],\n","                    [-7.5732612e-05, -1.7752903e-04,  2.9859928e-04, ...,\n","                     -4.4988643e-04, -7.6748733e-04,  4.5442476e-04],\n","                    ...,\n","                    [ 5.0231913e-04,  3.1881614e-04,  2.3118982e-03, ...,\n","                      2.0480863e-04, -3.3765702e-04,  4.0180862e-04],\n","                    [ 1.5452938e-04, -1.6484456e-04, -2.5985966e-04, ...,\n","                     -3.1373498e-04, -6.4456963e-04,  6.8054658e-05],\n","                    [ 4.4266126e-04, -5.3250132e-04,  2.0249723e-04, ...,\n","                      4.3182055e-04, -2.8395420e-04, -3.8511353e-05]], dtype=float32),\n","             array([-2.73065234e-04, -1.13433725e-05, -1.88623846e-04, ...,\n","                     4.09382519e-05, -1.28432084e-05, -8.24478411e-05], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[ 5.6972943e-04,  1.5091708e-04, -2.4517934e-04, ...,\n","                      2.5574718e-05, -8.2842234e-05, -3.8236030e-04],\n","                    [ 2.4575993e-04,  5.0802360e-04, -3.5157916e-04, ...,\n","                      6.7960040e-04, -7.1845221e-04, -7.0106715e-04],\n","                    [-4.0512768e-04,  2.0729403e-04,  8.0519175e-04, ...,\n","                     -2.3716423e-04, -9.7364059e-04,  5.5228232e-04],\n","                    ...,\n","                    [-8.4692991e-04,  2.0505050e-04,  1.8182152e-04, ...,\n","                     -7.2674465e-04,  1.2999673e-04, -9.7030192e-04],\n","                    [-4.4623425e-04, -4.5330101e-04,  2.7329684e-04, ...,\n","                      3.7517480e-04,  9.1525864e-05, -8.7958739e-05],\n","                    [ 4.3043011e-04,  2.7013093e-04, -1.0943448e-04, ...,\n","                     -2.4152960e-04, -4.0994296e-04,  4.7121712e-04]], dtype=float32),\n","             array([-3.0371919e-05,  7.2433919e-05, -4.3698977e-04, ...,\n","                     1.8775304e-04,  8.7853972e-05, -2.2629586e-04], dtype=float32)),\n","            ((), (), ())))),\n","         (),\n","         ((), (), ()),\n","         (array([[ 3.3750647e-04, -1.2777644e-04, -3.7869098e-04, ...,\n","                  -2.9960246e-04,  1.9728354e-06,  4.0162366e-04],\n","                 [ 4.3322399e-04,  1.7962696e-04, -5.1586493e-04, ...,\n","                   7.1843839e-05,  3.9571323e-04, -1.3421859e-04],\n","                 [ 5.7112065e-04,  9.9040772e-05,  8.5089581e-05, ...,\n","                   2.6148386e-04,  1.5803186e-04, -7.1445911e-04],\n","                 ...,\n","                 [ 4.9351162e-04,  5.9139676e-04,  8.5478555e-04, ...,\n","                   4.9972517e-04, -4.8057127e-04,  1.3736630e-04],\n","                 [ 3.7982041e-04, -2.0904833e-04,  9.2062262e-05, ...,\n","                  -9.7695598e-04, -7.4140895e-05,  4.7488933e-04],\n","                 [-1.4294167e-04,  4.3310088e-04, -1.3480733e-04, ...,\n","                   4.4059448e-06,  3.8522683e-04, -1.7768175e-04]], dtype=float32),\n","          array([-2.8041830e-05,  8.6050197e-05,  1.0145415e-05, ...,\n","                 -9.9558507e-05,  4.2972923e-05, -6.4754386e-05], dtype=float32))),\n","        ()),\n","       ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([25.133646, 20.16778 , 20.5209  , ..., 23.984125, 21.692732,\n","              23.309326], dtype=float32),\n","       array([-0.00082057, -0.00013815,  0.0003551 , ...,  0.00019342,\n","              -0.00048613, -0.00042382], dtype=float32)),\n","      (),\n","      ((), (((array([[-4.1568291e-04, -7.0044992e-04,  9.0338511e-04, ...,\n","                  -2.8851160e-04,  1.1637668e-04, -9.3022572e-05],\n","                 [-1.0806188e-04, -3.5761492e-04, -2.1240997e-04, ...,\n","                  -5.3244957e-04, -6.1152707e-05,  4.5347607e-04],\n","                 [-6.7972978e-05, -5.1062577e-04, -2.0633308e-04, ...,\n","                  -5.7725690e-04, -1.7183056e-04, -4.8175061e-04],\n","                 ...,\n","                 [-1.4764438e-05, -4.0215795e-04,  5.8179174e-04, ...,\n","                  -1.4387448e-03,  1.8242991e-03,  4.9638888e-04],\n","                 [-1.9918228e-03, -3.1295458e-05, -6.8738387e-04, ...,\n","                  -5.1219243e-04, -3.9047515e-04, -5.9968021e-05],\n","                 [ 8.7556452e-04, -1.5976807e-04,  1.0815831e-03, ...,\n","                   4.6251519e-04, -1.1048026e-03,  1.3263522e-03]], dtype=float32),\n","          array([ 0.00029522,  0.00055426, -0.00044644, ..., -0.00042406,\n","                  0.000437  ,  0.00022192], dtype=float32)),\n","         ((array([[-6.8464200e-05, -1.8233093e-04, -3.3943262e-04, ...,\n","                   -1.8391799e-04,  3.6906448e-04, -4.9532537e-04],\n","                  [ 9.7474369e-04,  4.3888067e-04,  1.8747739e-04, ...,\n","                   -9.8145450e-04,  2.7901010e-04, -1.3401183e-04],\n","                  [-8.1080398e-05, -1.1885613e-04, -3.8773671e-04, ...,\n","                    7.2938557e-05,  7.8095203e-05,  1.0366609e-04],\n","                  ...,\n","                  [-7.0196565e-04,  3.1755792e-04, -2.1898457e-04, ...,\n","                   -3.9910609e-04, -9.8375938e-05, -4.3949051e-04],\n","                  [ 1.2974661e-04,  1.5637414e-04, -5.3605152e-04, ...,\n","                   -1.3689074e-03,  2.5213658e-04, -3.8303563e-04],\n","                  [ 1.2799115e-03,  9.0032932e-04, -1.3062430e-03, ...,\n","                    2.8300483e-04,  8.9340011e-04, -4.1155572e-04]], dtype=float32),\n","           array([ 2.9129299e-04,  5.5194489e-04, -5.1632262e-04, ...,\n","                   1.3906155e-04, -4.8594858e-04,  1.7186281e-05], dtype=float32)),),\n","         ((array([[-2.5338688e-04, -5.4139073e-04,  7.4766623e-04, ...,\n","                    3.6963189e-04, -2.9117410e-04, -4.9463884e-05],\n","                  [-1.2523396e-04, -1.3393551e-04, -3.1212976e-04, ...,\n","                    8.9347345e-04,  3.4973022e-04,  2.0901949e-04],\n","                  [-1.6622439e-04, -1.5694088e-04,  1.0432038e-04, ...,\n","                    1.1873069e-04,  1.4051730e-04,  9.7769902e-05],\n","                  ...,\n","                  [ 3.0847662e-04,  2.9444607e-04,  2.8557068e-04, ...,\n","                    7.9267519e-04, -4.0208208e-04,  2.5114077e-04],\n","                  [ 5.5219902e-04, -1.7765032e-04,  6.4835498e-05, ...,\n","                    7.7855286e-05,  8.0095560e-05, -2.2911308e-04],\n","                  [-1.8453391e-04, -4.9205153e-04,  3.6849163e-04, ...,\n","                    1.6601085e-04, -8.7851560e-04, -2.8440345e-04]], dtype=float32),\n","           array([ 1.1125785e-04,  4.4743338e-05, -3.2781044e-04, ...,\n","                  -2.9616983e-04,  1.3335231e-04,  1.3582692e-04], dtype=float32)),)),\n","        (),\n","        (array([[-3.9797727e-04,  1.3177890e-04,  5.5765562e-05, ...,\n","                 -4.7081683e-04,  2.4590682e-04,  3.4581200e-04],\n","                [-5.8119162e-04,  2.4548584e-05, -2.1996321e-04, ...,\n","                 -3.7294815e-04, -1.3789561e-04, -3.6860383e-04],\n","                [ 1.3264106e-04, -5.1717955e-04, -5.0068187e-04, ...,\n","                 -1.6057723e-04,  5.8980635e-04, -2.4560481e-04],\n","                ...,\n","                [ 2.3485825e-04,  3.0704492e-04, -4.5639742e-04, ...,\n","                 -1.7103233e-04,  1.1021600e-04, -2.9392040e-04],\n","                [-9.6227610e-05, -1.7535933e-05, -8.4224725e-05, ...,\n","                  7.0292200e-04,  6.4043293e-04, -4.4276597e-04],\n","                [ 3.1093793e-04,  7.6029187e-06,  1.0981585e-05, ...,\n","                  5.9430290e-04, -1.9735850e-04,  2.1739390e-04]], dtype=float32),\n","         array([-1.4302914e-06,  9.7422189e-06,  1.0513639e-04, ...,\n","                -5.7261350e-05,  1.6653375e-05,  5.1994084e-05], dtype=float32))), ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([19.214445, 20.197676, 16.931206, ..., 22.679222, 16.830877,\n","              26.278265], dtype=float32),\n","       array([-0.0025097 , -0.00190769,  0.00266803, ..., -0.0026909 ,\n","              -0.0025036 , -0.00083268], dtype=float32)),\n","      (array([[ 1.2854071e-04, -1.4049307e-04,  5.7212741e-04, ...,\n","                4.6499239e-04, -1.8449577e-04, -5.0291361e-04],\n","              [-4.3347629e-04, -2.8481951e-04, -1.8057977e-04, ...,\n","               -6.7624074e-05,  2.8334003e-05, -1.8753945e-04],\n","              [-3.9501503e-04,  7.2625879e-04, -6.5977959e-04, ...,\n","                1.5979448e-04, -5.9743656e-04, -2.4528228e-04],\n","              ...,\n","              [-2.1153205e-04, -3.4640235e-04,  2.6383251e-04, ...,\n","               -1.0582620e-04,  2.7256153e-04, -7.2117423e-04],\n","              [-3.9072646e-04, -4.7944006e-04,  1.2331836e-04, ...,\n","               -7.5250719e-04,  9.4384479e-04,  4.9761293e-04],\n","              [-8.5363143e-05, -5.5270677e-04,  5.7721982e-04, ...,\n","                6.0686574e-04, -1.9075462e-04,  6.8411196e-04]], dtype=float32),\n","       array([-0.000316  , -0.00029598, -0.00019718, ..., -0.00025396,\n","              -0.000149  , -0.000182  ], dtype=float32)),\n","      ((), (), ()),\n","      (),\n","      (array([[-1.5870156e-04, -1.7582296e-04,  9.9765872e-05, ...,\n","               -6.9899332e-05,  4.7058155e-04, -1.0766292e-03],\n","              [-4.8029472e-04,  5.4609368e-04, -4.8330455e-04, ...,\n","                2.9035346e-05,  9.5068343e-04, -5.3134485e-05],\n","              [-5.4901605e-04,  3.6975666e-04, -2.2262198e-04, ...,\n","               -7.2464487e-04,  8.3791633e-04, -3.4418871e-04],\n","              ...,\n","              [ 1.2364995e-04,  8.3441165e-04,  4.3902599e-04, ...,\n","                4.6774247e-04,  1.7298660e-05, -2.4150473e-04],\n","              [-6.2846747e-04,  5.1117555e-04, -6.9211778e-04, ...,\n","                3.3828150e-04,  6.5276399e-05, -1.3280950e-03],\n","              [ 3.1406962e-04,  3.8795883e-04, -1.2104869e-04, ...,\n","                7.2453707e-04,  7.3430303e-04, -7.1827089e-04]], dtype=float32),\n","       array([-1.90261533e-04,  1.19103526e-04,  7.10676322e-05, ...,\n","               1.85482240e-05,  1.88032151e-04, -1.35134032e-05], dtype=float32)),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([19.002293, 19.749205, 20.942585, ..., 24.964912, 20.48379 ,\n","              20.58046 ], dtype=float32),\n","       array([-5.3074583e-04,  4.3791084e-04,  1.9000332e-04, ...,\n","              -7.0100659e-06,  4.9053360e-04,  1.5993371e-04], dtype=float32)),\n","      ((),\n","       ((),\n","        (((), (((array([[-3.2662504e-04,  1.1444593e-04,  2.4137925e-04, ...,\n","                     -1.0646781e-03,  9.6945019e-05, -7.5971446e-04],\n","                    [ 5.0910490e-05, -6.4728607e-04,  7.2481611e-04, ...,\n","                     -3.9798219e-04, -4.0684509e-04,  3.8374279e-04],\n","                    [-2.4369388e-04, -2.4113182e-05, -1.6093491e-04, ...,\n","                      1.7865638e-03, -2.9024642e-04, -5.2316190e-04],\n","                    ...,\n","                    [ 3.4718850e-04,  5.6409038e-04, -4.2735969e-04, ...,\n","                     -1.2790303e-03, -2.1404806e-03, -3.2736691e-05],\n","                    [ 4.4267022e-04,  7.6589599e-04,  2.4049124e-04, ...,\n","                     -1.5362485e-04,  6.3673867e-04,  2.9148831e-07],\n","                    [ 5.9032941e-04, -1.5744078e-04,  3.5832392e-04, ...,\n","                     -1.1176147e-03, -5.0487573e-04, -1.4310669e-03]], dtype=float32),\n","             array([ 2.4307283e-04, -7.5164338e-05,  3.6123133e-04, ...,\n","                     1.4516230e-04,  6.7123206e-04,  2.9318323e-04], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[ 1.1629520e-04,  8.8479707e-04, -1.9393528e-04, ...,\n","                      4.7091907e-04,  9.6876203e-04,  1.0016921e-03],\n","                    [ 2.5273944e-04, -2.5784405e-04,  1.1666737e-03, ...,\n","                      1.3402230e-03,  1.7075094e-04,  4.8120902e-04],\n","                    [ 2.3882565e-04, -1.2218894e-03, -6.2975992e-04, ...,\n","                      5.5051252e-04,  5.5383286e-04, -7.6446915e-04],\n","                    ...,\n","                    [ 8.3121115e-05,  9.4296538e-06,  5.7837591e-05, ...,\n","                     -7.0200645e-04,  7.9577835e-04, -1.1063863e-03],\n","                    [ 2.7475145e-04, -3.2025401e-04,  2.4247781e-04, ...,\n","                      2.5902310e-04,  1.8698905e-03,  4.3908472e-04],\n","                    [ 2.8282005e-04, -6.4120494e-04, -4.8135631e-04, ...,\n","                      3.1380585e-04,  2.6316915e-04,  4.4861110e-04]], dtype=float32),\n","             array([ 0.00040362, -0.00071073,  0.00029252, ..., -0.00015833,\n","                    -0.00023575, -0.00052007], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[-2.8662401e-04, -9.4027986e-05, -2.1397236e-06, ...,\n","                      1.9451615e-04, -2.3048274e-04,  8.4594707e-04],\n","                    [ 1.0381620e-04,  3.7930735e-05,  6.8526412e-04, ...,\n","                      1.2551635e-04,  6.3297641e-04, -6.2677776e-04],\n","                    [-9.3176117e-05, -2.0888013e-04,  2.8213556e-04, ...,\n","                     -9.1187470e-04,  3.9832263e-05, -1.2100150e-03],\n","                    ...,\n","                    [ 2.9264446e-04,  3.2617070e-04, -4.2719385e-04, ...,\n","                     -1.3374632e-03,  5.0895510e-04, -1.2233338e-03],\n","                    [-3.8262631e-04,  1.7408621e-04,  7.0922011e-05, ...,\n","                      1.6274179e-03,  1.4907118e-03, -6.3161779e-04],\n","                    [-4.8971815e-05,  1.1881483e-04,  1.3580993e-06, ...,\n","                     -7.6949911e-04,  3.3165538e-04,  6.6358701e-04]], dtype=float32),\n","             array([ 1.4807894e-05, -1.5469561e-05,  1.7480261e-04, ...,\n","                    -1.4047622e-04, -6.3886691e-05,  2.5733566e-04], dtype=float32)),\n","            ((), (), ())))),\n","         (),\n","         ((), (), ()),\n","         (array([[ 4.19790158e-04, -1.88334510e-04,  1.17033596e-04, ...,\n","                  -7.74421496e-05,  2.69945886e-04, -3.33117059e-04],\n","                 [ 3.97198077e-04, -1.01683590e-05,  2.84180365e-04, ...,\n","                  -2.04462151e-04,  5.11663093e-04,  2.75116836e-05],\n","                 [-8.69778960e-05, -7.26998725e-04,  1.16389110e-05, ...,\n","                   5.58125728e-04,  2.83130412e-05,  3.58814796e-05],\n","                 ...,\n","                 [-8.19593261e-04, -1.59495440e-03, -1.85846293e-04, ...,\n","                   5.41982590e-04, -2.67938187e-04, -4.86780802e-04],\n","                 [-2.95832666e-04,  5.99099349e-06, -3.82576371e-04, ...,\n","                  -4.29838314e-04,  3.25089030e-04, -4.65002347e-04],\n","                 [-2.88312061e-04, -3.01397318e-04,  1.35712384e-03, ...,\n","                   2.06341574e-04,  8.55413324e-04, -2.74819817e-04]], dtype=float32),\n","          array([-5.05392127e-05, -1.90757884e-04,  1.16461095e-04, ...,\n","                 -8.70836593e-05, -1.01621881e-04,  1.90394494e-04], dtype=float32))),\n","        ()),\n","       ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([23.863585, 23.908377, 22.176037, ..., 20.800194, 23.99026 ,\n","              20.214481], dtype=float32),\n","       array([-0.00095918, -0.00016006,  0.00014331, ...,  0.000334  ,\n","              -0.00060179, -0.00046619], dtype=float32)),\n","      (),\n","      ((), (((array([[-5.6311046e-04, -3.4005710e-04, -5.9890590e-04, ...,\n","                  -7.4382929e-06, -3.6587298e-04, -1.0307857e-03],\n","                 [ 1.0773811e-04,  4.9511186e-04,  3.1247773e-04, ...,\n","                   7.7393785e-04,  1.7064218e-04,  1.3642012e-04],\n","                 [-5.8631634e-04,  7.4155215e-04, -2.4314056e-04, ...,\n","                  -6.5048553e-05,  7.3810411e-04, -7.5508637e-04],\n","                 ...,\n","                 [ 5.2821374e-04,  7.6992292e-04, -1.0839757e-04, ...,\n","                  -4.1444082e-04,  6.1353593e-04, -6.9915073e-04],\n","                 [ 5.5921635e-05,  8.9419846e-05,  1.4359289e-03, ...,\n","                   5.2405300e-04, -6.8278838e-04,  6.0381956e-04],\n","                 [ 9.9906547e-04, -3.9901065e-06,  6.5774639e-04, ...,\n","                   2.5885552e-04,  1.1800595e-03,  2.0022877e-04]], dtype=float32),\n","          array([ 1.61186490e-05, -3.49949289e-04,  4.99187299e-05, ...,\n","                  1.76321100e-06, -1.09602806e-04, -1.44498617e-05], dtype=float32)),\n","         ((array([[ 1.4999622e-03, -1.1105599e-03,  7.6823495e-04, ...,\n","                   -4.1793473e-04, -9.6628646e-04,  5.2762109e-05],\n","                  [ 7.6032698e-04,  4.4384800e-04, -8.5847086e-04, ...,\n","                   -3.9488904e-04,  1.5121815e-04,  5.3509368e-05],\n","                  [-6.5694255e-04, -5.0702689e-05, -2.1640853e-04, ...,\n","                    2.4654322e-05,  2.0967561e-04, -7.6775131e-04],\n","                  ...,\n","                  [-4.8284343e-04,  6.5310975e-04, -6.8984577e-04, ...,\n","                    6.1466306e-04,  1.3914340e-03, -6.9069950e-04],\n","                  [ 9.7297557e-04, -5.4459364e-05,  1.7869713e-03, ...,\n","                   -1.9494690e-04,  1.4137512e-04,  5.1790511e-04],\n","                  [-2.2960478e-05,  4.4922695e-06,  7.2395243e-04, ...,\n","                   -5.3702993e-04,  1.7198462e-04,  4.2873158e-04]], dtype=float32),\n","           array([ 7.2075153e-04,  9.1769427e-05,  3.2753008e-04, ...,\n","                  -1.8249797e-04, -3.9868570e-05, -4.1253286e-05], dtype=float32)),),\n","         ((array([[ 7.56191912e-06,  3.52023984e-04, -1.45454751e-05, ...,\n","                    1.36408298e-05,  1.03212864e-04, -4.69582883e-04],\n","                  [-2.17174151e-04, -1.32270812e-04,  3.02658678e-04, ...,\n","                    4.18907061e-04,  4.74628527e-04, -4.15859322e-05],\n","                  [ 2.28795107e-04,  3.33249540e-04, -9.53869239e-05, ...,\n","                   -6.97138719e-04,  1.65113015e-04,  4.89227765e-04],\n","                  ...,\n","                  [ 3.71309470e-05,  1.23884583e-05,  9.71424379e-05, ...,\n","                   -4.36817645e-05, -1.02943904e-05, -1.76644768e-04],\n","                  [-3.76037438e-04, -5.35300758e-04, -5.71178389e-04, ...,\n","                   -5.22463142e-06, -3.68841975e-05, -7.98741821e-05],\n","                  [-3.91982263e-04, -8.13209263e-05,  1.27167266e-04, ...,\n","                    2.55005376e-04, -5.68897056e-04,  4.26564511e-04]], dtype=float32),\n","           array([ 4.7820471e-05,  1.5647970e-04,  2.6233769e-05, ...,\n","                   3.1863447e-04,  1.2342793e-04, -3.4023362e-04], dtype=float32)),)),\n","        (),\n","        (array([[ 1.21238125e-04, -5.54900034e-04,  1.05623119e-04, ...,\n","                  3.35781224e-04, -3.78645782e-05, -8.33227299e-04],\n","                [-5.29213808e-04, -5.15382490e-05,  6.98506992e-05, ...,\n","                 -7.41509255e-04,  1.74591914e-04, -4.21049132e-04],\n","                [ 1.20665427e-04, -1.02331885e-03, -2.44848197e-04, ...,\n","                  1.52682551e-04, -4.46544611e-04, -2.26254604e-04],\n","                ...,\n","                [ 6.35489938e-04, -3.70408998e-05, -4.66282101e-04, ...,\n","                  4.34915215e-04, -1.95286528e-04,  9.23890970e-04],\n","                [-3.32740252e-04, -4.04429680e-04,  3.48495159e-05, ...,\n","                  9.20497812e-04, -1.48475774e-05,  7.63617107e-04],\n","                [ 4.12746507e-04, -3.16594582e-04,  2.39612738e-04, ...,\n","                  6.49387483e-04,  6.79656514e-04,  5.86012262e-04]], dtype=float32),\n","         array([-4.2105734e-05, -8.8575449e-05,  2.1887388e-05, ...,\n","                 9.4275601e-05,  1.5780930e-05, -4.0148479e-05], dtype=float32))), ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([19.34309 , 16.7217  , 18.787088, ..., 24.8799  , 16.975502,\n","              29.151562], dtype=float32),\n","       array([ 0.00056903, -0.00049533, -0.00038546, ..., -0.00245594,\n","              -0.00227448, -0.00023161], dtype=float32)),\n","      (array([[ 1.4499012e-04, -4.0405830e-06,  9.2359469e-04, ...,\n","               -4.0990015e-04, -6.4596687e-05, -4.3329134e-04],\n","              [ 5.1942200e-04,  4.3853107e-04, -3.3276502e-04, ...,\n","                4.6164618e-04,  2.1372586e-04, -4.4197761e-04],\n","              [ 2.9106968e-04,  1.4575070e-04,  1.5153109e-04, ...,\n","                2.1366282e-04, -2.4357258e-05, -3.6520240e-04],\n","              ...,\n","              [ 7.4607633e-05, -3.2630228e-04,  3.7390104e-04, ...,\n","               -2.7619544e-04, -1.1030681e-04,  3.7435620e-04],\n","              [-8.9861365e-05,  1.3448064e-04,  7.4247102e-04, ...,\n","                8.1414881e-05, -1.3266108e-04,  2.3058454e-05],\n","              [ 7.1145804e-04,  1.9146790e-04, -2.7287356e-04, ...,\n","                6.8428437e-04,  1.4821423e-04, -6.7117762e-05]], dtype=float32),\n","       array([-2.3738350e-04, -2.0401976e-04, -1.7334563e-04, ...,\n","              -2.4547981e-04, -2.9552355e-04, -9.8087723e-05], dtype=float32)),\n","      ((), (), ()),\n","      (),\n","      (array([[-5.67895622e-05, -2.73075508e-04, -6.24363718e-04, ...,\n","               -4.14357710e-05, -1.29626031e-04,  6.19488943e-04],\n","              [ 2.25377575e-04, -1.72959535e-05,  2.15629581e-04, ...,\n","               -7.46554404e-04, -7.59310729e-04, -1.91541316e-04],\n","              [-3.76099226e-04,  4.92840132e-04, -3.14589910e-04, ...,\n","               -1.23628764e-04, -1.61852819e-04, -4.53198390e-06],\n","              ...,\n","              [-8.65741968e-05, -3.11507552e-04,  4.54512367e-04, ...,\n","                8.42900481e-04,  6.83342747e-04, -3.23505665e-04],\n","              [-3.85042396e-04, -2.67523370e-04,  5.21010312e-04, ...,\n","               -6.93354872e-04, -2.59882290e-05,  2.99303822e-04],\n","              [ 8.01490460e-05,  2.38499852e-04, -5.64297952e-04, ...,\n","               -5.83622837e-04,  1.11495116e-04, -2.78797990e-04]], dtype=float32),\n","       array([-1.9297891e-04, -2.6990603e-05,  2.1672313e-04, ...,\n","               2.3786057e-05,  5.6566038e-05,  2.0984407e-05], dtype=float32)),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([25.713665, 20.014107, 24.006456, ..., 19.69559 , 18.432018,\n","              22.437933], dtype=float32),\n","       array([-0.00034028,  0.00051509,  0.00037291, ...,  0.00017522,\n","               0.00065555,  0.00034202], dtype=float32)),\n","      ((),\n","       ((),\n","        (((),\n","          (((array([[-5.95373858e-04, -6.70322101e-04,  6.84015802e-04, ...,\n","                      1.01916713e-03,  3.06264119e-04, -2.12282146e-04],\n","                    [-1.09707902e-03,  4.93289146e-04,  3.01994623e-05, ...,\n","                     -6.51322363e-04, -1.20606164e-04, -7.97604444e-04],\n","                    [-3.34893237e-04,  2.98915311e-05,  1.31439138e-03, ...,\n","                     -9.88212763e-04,  1.58660836e-03,  1.05705221e-04],\n","                    ...,\n","                    [ 3.71588016e-04, -3.57395184e-04,  6.24878099e-04, ...,\n","                      7.03756348e-04, -9.69733403e-04,  2.75732542e-04],\n","                    [-5.07866556e-04,  3.22071166e-04, -9.61839396e-04, ...,\n","                     -4.04917955e-04, -9.88954562e-05,  8.26840464e-04],\n","                    [-4.99062066e-04,  7.66336962e-06, -7.86092016e-04, ...,\n","                      1.45599688e-05,  2.81904067e-04, -7.28202576e-04]], dtype=float32),\n","             array([-0.00033887,  0.00083491,  0.00049547, ..., -0.00046566,\n","                     0.00038666,  0.00023574], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[-4.6708356e-04,  6.9332297e-04,  7.5118773e-04, ...,\n","                     -5.6861638e-04,  5.6659104e-04, -4.1663705e-04],\n","                    [ 6.3709583e-04,  6.7082909e-04,  3.6681963e-06, ...,\n","                      3.0855057e-04, -4.8214087e-04,  6.0388952e-04],\n","                    [-5.2432850e-04, -4.2004426e-04, -6.7650905e-04, ...,\n","                      2.4281103e-04,  1.2098042e-03,  1.7371315e-03],\n","                    ...,\n","                    [-1.6805739e-04, -1.8724446e-04, -4.7246023e-04, ...,\n","                     -1.2733364e-03,  6.1976556e-05, -1.9401417e-04],\n","                    [ 3.2186403e-04, -6.6762610e-04, -1.8793202e-04, ...,\n","                      5.0563191e-04,  3.6959755e-05,  2.6919985e-05],\n","                    [ 5.8454339e-04,  1.1800550e-04,  2.8509271e-04, ...,\n","                      1.9577749e-04, -4.8818914e-05, -5.7708053e-04]], dtype=float32),\n","             array([ 8.7273773e-05, -1.6973160e-04, -2.7384833e-05, ...,\n","                     1.0870067e-04,  3.8789283e-04, -9.8721262e-05], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[ 1.4023768e-04,  1.7267250e-05,  1.1397027e-04, ...,\n","                      3.3457161e-04, -9.3385199e-04,  2.6291612e-04],\n","                    [-4.7741775e-04, -3.5663110e-05, -4.7508275e-04, ...,\n","                      1.8410153e-04, -6.5509131e-04, -3.7956273e-04],\n","                    [ 2.6665510e-05,  9.7706637e-05, -7.2879094e-04, ...,\n","                      2.0608396e-04,  4.0892858e-04, -6.1673491e-05],\n","                    ...,\n","                    [-5.9672334e-04, -7.7638635e-04, -1.7216546e-04, ...,\n","                     -2.6407541e-04,  7.3876500e-04, -3.8623079e-04],\n","                    [ 1.6931896e-04, -6.6302647e-04, -5.1263312e-04, ...,\n","                      5.6379085e-04, -3.7311375e-04, -4.0221211e-04],\n","                    [ 5.0163362e-04, -4.2826321e-04,  9.6582022e-04, ...,\n","                     -1.2359548e-04,  9.6388096e-05,  6.5588980e-04]], dtype=float32),\n","             array([ 1.4728331e-04, -3.4281524e-04, -2.8725539e-04, ...,\n","                    -9.9542140e-06,  2.3434323e-04,  3.0246750e-04], dtype=float32)),\n","            ((), (), ())))),\n","         (),\n","         ((), (), ()),\n","         (array([[ 6.6545577e-04, -2.9493417e-04, -1.6901178e-05, ...,\n","                  -3.0363153e-04, -8.7133789e-04,  8.0927910e-04],\n","                 [-5.0262583e-04,  5.7844073e-04, -1.9344705e-04, ...,\n","                   1.0666212e-03, -5.9937389e-04, -4.5814467e-04],\n","                 [-6.2915392e-04,  1.7435101e-04, -1.0406631e-03, ...,\n","                   2.8984260e-04,  2.6763559e-04,  1.4070586e-04],\n","                 ...,\n","                 [ 1.2310820e-04, -4.0598653e-04, -1.3826853e-04, ...,\n","                   3.3614406e-04, -4.4232060e-04,  2.3486126e-04],\n","                 [-1.3146414e-04, -5.8402517e-04, -5.8737722e-05, ...,\n","                  -2.8182194e-04, -1.5414132e-04,  1.2118524e-04],\n","                 [ 2.4960178e-04,  1.5976763e-04, -1.2432698e-04, ...,\n","                   8.7708562e-05, -3.9536858e-04,  3.0887790e-05]], dtype=float32),\n","          array([ 1.08676933e-04, -1.39384108e-04,  1.54571171e-04, ...,\n","                 -1.89793573e-05, -2.18613655e-04,  1.04364655e-04], dtype=float32))),\n","        ()),\n","       ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([18.377483, 20.529018, 15.54321 , ..., 13.164674, 16.287607,\n","              19.060873], dtype=float32),\n","       array([-1.1201794e-03,  8.3273895e-05, -2.0866966e-05, ...,\n","              -2.0652132e-04, -3.9776845e-04, -4.0178094e-04], dtype=float32)),\n","      (),\n","      ((), (((array([[-4.2494872e-04,  4.0700205e-04,  1.0334866e-03, ...,\n","                   1.7160259e-04,  8.2796352e-04,  1.2134418e-04],\n","                 [-5.3424540e-04,  2.0468986e-04, -9.8973524e-04, ...,\n","                  -4.6759282e-04,  5.1914080e-04,  3.9970566e-04],\n","                 [ 9.5913289e-05,  4.8852043e-04,  1.6801392e-04, ...,\n","                   7.2381017e-04, -7.8206672e-04, -4.7479474e-04],\n","                 ...,\n","                 [-6.2540680e-04, -6.9601816e-04, -7.4346882e-04, ...,\n","                  -2.8362262e-04,  6.3001452e-04,  5.6671828e-04],\n","                 [ 1.9049420e-04,  4.3764294e-04, -2.6377838e-04, ...,\n","                   2.0465213e-04,  1.1279173e-05, -1.0332718e-03],\n","                 [-8.8347579e-06,  4.7042215e-04,  1.1780338e-03, ...,\n","                  -6.1488559e-04,  7.7360921e-05,  3.1986329e-04]], dtype=float32),\n","          array([ 1.8846759e-04,  2.5323729e-04, -3.4402986e-04, ...,\n","                  6.1463536e-05, -9.5363648e-05, -5.8558937e-05], dtype=float32)),\n","         ((array([[-2.20481597e-04,  1.18063518e-03,  1.56215043e-03, ...,\n","                    9.82606216e-05,  1.44002755e-04,  9.13276977e-04],\n","                  [-1.19752530e-03,  2.23923766e-04,  4.35432623e-04, ...,\n","                   -2.78261665e-04,  3.18312785e-04,  4.21736739e-04],\n","                  [ 2.22250237e-04, -1.60400843e-04, -1.08029635e-04, ...,\n","                    1.12295791e-03, -5.08838675e-05, -7.32662040e-04],\n","                  ...,\n","                  [-4.19384305e-04,  1.50420790e-04, -3.58337129e-05, ...,\n","                    9.65280633e-04, -6.97434007e-04, -3.43855441e-04],\n","                  [-3.67292487e-05, -3.13550234e-04,  5.65835740e-04, ...,\n","                    4.96019318e-04, -2.73407291e-04,  1.34612244e-04],\n","                  [-2.06746015e-04,  1.40402687e-03, -1.46131148e-03, ...,\n","                   -6.88987842e-04,  7.68021157e-04,  4.77079797e-04]], dtype=float32),\n","           array([ 0.00045397, -0.00019233, -0.00010555, ...,  0.00019466,\n","                   0.0005841 ,  0.00033662], dtype=float32)),),\n","         ((array([[-5.7503395e-04, -8.6632353e-05, -1.2176593e-04, ...,\n","                   -4.1146272e-05, -5.1465853e-05, -1.2289004e-04],\n","                  [-5.7021934e-05, -1.7840607e-04, -5.6992569e-05, ...,\n","                    3.3710475e-04,  6.7078992e-04,  4.5398093e-04],\n","                  [ 3.6924469e-04, -2.1842278e-04, -3.4643107e-04, ...,\n","                    3.2703512e-04,  3.7312473e-04,  6.6157495e-06],\n","                  ...,\n","                  [ 2.3037243e-04,  1.2348396e-05,  6.5281370e-04, ...,\n","                   -3.6747540e-06,  7.1686452e-05,  2.1650737e-04],\n","                  [-5.9988169e-04, -3.6153905e-04, -3.9566023e-04, ...,\n","                   -4.1974065e-04, -1.0731411e-04,  4.9347471e-04],\n","                  [-5.1677995e-04, -4.0979515e-04, -3.2814703e-04, ...,\n","                    1.7364566e-04, -3.7668971e-04,  2.6420257e-05]], dtype=float32),\n","           array([ 1.2099383e-04, -1.6186872e-05, -9.6671807e-05, ...,\n","                  -8.6190557e-05,  1.6071007e-04,  1.2687828e-04], dtype=float32)),)),\n","        (),\n","        (array([[-5.2951230e-04, -3.0873933e-05, -4.3894153e-04, ...,\n","                 -4.5457709e-05,  1.6933204e-04, -3.2049805e-04],\n","                [-7.6115213e-04,  1.0051356e-03,  1.0210091e-05, ...,\n","                 -5.8094744e-04,  3.4703006e-04,  2.4947929e-04],\n","                [-2.5880482e-04,  4.9121049e-04, -2.4654556e-04, ...,\n","                  3.3488730e-05,  3.9486616e-04, -1.9640291e-04],\n","                ...,\n","                [-2.3098798e-04, -5.4062082e-04, -2.6964777e-04, ...,\n","                 -2.3456400e-04,  4.3855066e-04,  4.0015514e-04],\n","                [-2.2577356e-04, -3.0327970e-04,  1.7653166e-04, ...,\n","                 -1.2154292e-04, -1.7773600e-04, -4.3495081e-04],\n","                [-4.9273425e-04, -5.3295371e-05,  4.2104273e-04, ...,\n","                 -3.8977829e-04, -1.1681722e-04, -4.6599929e-05]], dtype=float32),\n","         array([ 9.0977250e-05, -2.2555199e-04,  1.3827017e-04, ...,\n","                 1.2001016e-04, -1.3514365e-04,  1.6845731e-04], dtype=float32))), ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([28.375448, 21.047567, 26.7278  , ..., 36.833595, 28.114101,\n","              38.829243], dtype=float32),\n","       array([-0.00050365, -0.00304625, -0.00138255, ..., -0.00156267,\n","              -0.00550474, -0.00064232], dtype=float32)),\n","      (array([[ 4.51025408e-05,  2.89572752e-04,  7.18691153e-04, ...,\n","               -8.20976798e-04, -2.11006089e-04, -6.32703814e-05],\n","              [-2.03617878e-04,  1.46011953e-04,  8.05886011e-05, ...,\n","                4.29164065e-04, -2.71452882e-04,  4.74325177e-04],\n","              [ 3.37896199e-06,  4.69583174e-05, -2.14861939e-06, ...,\n","               -1.09011882e-04,  5.96934813e-04, -3.31864954e-04],\n","              ...,\n","              [-4.10760636e-04,  3.32822819e-04,  3.46525572e-04, ...,\n","                7.53040658e-04, -1.18421587e-04,  3.52360163e-04],\n","              [ 4.15057351e-04,  4.05785686e-04, -1.82250471e-04, ...,\n","                1.20039505e-04, -3.75431933e-04, -2.42003560e-04],\n","              [-3.49384471e-04,  1.53896603e-04, -1.97059999e-04, ...,\n","                3.18604230e-04,  7.52049382e-04, -3.79994308e-04]], dtype=float32),\n","       array([-0.00031323, -0.00026395, -0.00019575, ..., -0.00027545,\n","              -0.00012683, -0.00028225], dtype=float32)),\n","      ((), (), ()),\n","      (),\n","      (array([[-1.45592698e-04,  2.35519430e-04,  3.57860874e-04, ...,\n","               -1.53046945e-04, -2.44563271e-04, -1.83074386e-04],\n","              [ 1.27271007e-04,  3.07627168e-04,  1.10893976e-04, ...,\n","                4.36779635e-04, -5.40185894e-04, -4.86610923e-04],\n","              [ 2.46395502e-05,  4.56353475e-04, -6.73503091e-04, ...,\n","                9.86543717e-04, -1.95097295e-04, -3.08188348e-04],\n","              ...,\n","              [ 2.98702013e-04, -1.93927859e-04,  2.21260503e-04, ...,\n","                1.06906649e-04,  4.20694065e-04,  5.91091579e-04],\n","              [-3.40614526e-04, -1.57316917e-05,  1.02289378e-04, ...,\n","               -3.43150932e-05,  3.79157107e-04,  1.55710921e-04],\n","              [ 5.24621108e-04, -3.03163397e-04,  1.98939248e-04, ...,\n","               -4.11672454e-06, -4.95930726e-04, -2.03308577e-04]], dtype=float32),\n","       array([-1.7706344e-04, -6.3018502e-05,  1.9138168e-04, ...,\n","               2.3274153e-04, -1.2326252e-04, -6.7591194e-05], dtype=float32)),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([13.746678, 16.561274, 14.163938, ..., 13.718825, 10.817346,\n","              14.106273], dtype=float32),\n","       array([0.00021479, 0.00034326, 0.00048562, ..., 0.00017553, 0.00035025,\n","              0.00022324], dtype=float32)),\n","      ((),\n","       ((),\n","        (((), (((array([[-5.9680853e-05, -5.9021910e-04,  3.4338131e-04, ...,\n","                     -5.7301525e-04,  1.6458239e-04, -6.4730586e-05],\n","                    [ 9.0298936e-04,  3.9069095e-05, -8.8863348e-04, ...,\n","                     -2.4318568e-04, -6.1069272e-04, -7.5299229e-04],\n","                    [-4.9093876e-05,  1.1434063e-03,  6.8125402e-04, ...,\n","                      1.1330675e-06,  9.5477815e-05,  9.9931189e-05],\n","                    ...,\n","                    [-6.5700233e-04, -3.9993276e-04,  7.8064314e-04, ...,\n","                      7.4518140e-04,  1.3073550e-03,  6.2987983e-04],\n","                    [-2.8965160e-05, -1.6443384e-03, -4.4062926e-04, ...,\n","                      6.8620226e-04,  1.0453523e-03, -1.6417885e-04],\n","                    [-2.2878405e-04,  4.4170758e-04,  1.2165980e-03, ...,\n","                      4.2355640e-04,  3.3953482e-05, -1.9234538e-03]], dtype=float32),\n","             array([ 7.4027397e-04,  4.3148193e-05,  3.3105447e-05, ...,\n","                    -2.1822771e-04,  3.7661270e-05, -1.0889303e-03], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[ 1.9507897e-03, -5.3737592e-04, -4.0137104e-04, ...,\n","                     -8.6143857e-04,  2.9981211e-05, -1.3678310e-03],\n","                    [ 2.7808403e-06, -9.8241493e-04, -1.9275505e-03, ...,\n","                     -3.5545666e-04, -5.7414605e-04,  1.1727281e-04],\n","                    [-6.9833634e-04, -3.9387684e-04, -4.8862607e-04, ...,\n","                     -4.7911407e-04,  6.3502236e-04, -1.1824396e-04],\n","                    ...,\n","                    [ 2.1800643e-04, -4.9437530e-04, -1.5264755e-03, ...,\n","                     -8.9897524e-04, -5.7805527e-04,  4.6240803e-04],\n","                    [ 7.0580171e-04,  4.8551429e-04,  4.4714194e-04, ...,\n","                      3.0905585e-04, -1.4898789e-04,  1.7916671e-03],\n","                    [ 2.4140286e-04,  8.1153812e-05,  6.6683418e-04, ...,\n","                     -5.5251067e-04, -1.9300265e-03, -4.7680669e-04]], dtype=float32),\n","             array([ 0.00047172,  0.00023644,  0.0001964 , ...,  0.00017636,\n","                    -0.00029696, -0.00010085], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[ 2.8333615e-04,  2.6514224e-04,  2.8986830e-04, ...,\n","                     -9.0217701e-04, -1.7606845e-04,  3.5522593e-04],\n","                    [-2.9748355e-04, -5.8489229e-04, -8.3266222e-04, ...,\n","                     -7.8291696e-04,  6.3296466e-04,  1.8974216e-04],\n","                    [-4.4652153e-04,  1.0504105e-04,  4.5307967e-04, ...,\n","                     -1.1461754e-04, -7.5816737e-05, -5.0703038e-05],\n","                    ...,\n","                    [-9.1124984e-04,  4.2180796e-04,  4.5467366e-04, ...,\n","                     -1.6733776e-04,  2.6973825e-05,  8.1089651e-04],\n","                    [ 1.6803837e-05,  6.8659277e-04, -7.5581804e-05, ...,\n","                      2.0891393e-04, -1.8082996e-04, -9.3214228e-05],\n","                    [-6.3421414e-04,  3.7820791e-04, -3.9024602e-04, ...,\n","                      4.4681123e-04, -4.5484092e-05,  8.0207497e-04]], dtype=float32),\n","             array([ 1.65373800e-04,  1.95919056e-04, -3.30810843e-04, ...,\n","                    -6.23782544e-05, -7.75424924e-05, -1.10901565e-04], dtype=float32)),\n","            ((), (), ())))),\n","         (),\n","         ((), (), ()),\n","         (array([[-5.26038173e-04, -8.51672201e-04,  7.06742459e-04, ...,\n","                   6.67757820e-04,  1.26216677e-04,  3.41823616e-04],\n","                 [-5.70754928e-04,  7.32055225e-04, -3.95717856e-04, ...,\n","                  -3.39508755e-04, -5.80741907e-04, -9.59180237e-04],\n","                 [-2.88212177e-04,  5.54184779e-04,  2.86419963e-04, ...,\n","                  -1.93430591e-04,  2.38587876e-04,  1.14398776e-04],\n","                 ...,\n","                 [-4.67045262e-04, -1.18031458e-04,  6.06363814e-04, ...,\n","                   1.10043853e-04,  5.29236684e-04, -4.62355383e-04],\n","                 [-6.73850649e-04,  7.52628257e-04,  3.05649330e-04, ...,\n","                  -8.36021907e-04,  5.87670766e-05,  3.05817492e-04],\n","                 [ 1.00220670e-03,  6.73474511e-04, -1.61880831e-04, ...,\n","                   6.86252315e-04, -7.65125209e-04,  5.64046903e-04]], dtype=float32),\n","          array([-4.3577285e-05, -9.8750999e-05,  1.9142646e-04, ...,\n","                  6.5498816e-06,  6.8141708e-07,  2.2056769e-04], dtype=float32))),\n","        ()),\n","       ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([23.69895 , 22.030571, 19.21732 , ..., 14.703958, 20.725721,\n","              16.226542], dtype=float32),\n","       array([-6.9358299e-04,  8.2193961e-05,  3.5426844e-04, ...,\n","               5.0964102e-04, -9.3821931e-05, -5.8457587e-04], dtype=float32)),\n","      (),\n","      ((), (((array([[-1.8446751e-04,  3.8230038e-04,  7.4820593e-04, ...,\n","                  -4.2909558e-04,  1.7076760e-04, -3.2859368e-04],\n","                 [ 7.2607817e-04,  7.1218587e-04, -5.4862938e-04, ...,\n","                   3.9089471e-04, -7.6569646e-05,  5.6238170e-04],\n","                 [ 2.9848906e-04,  6.8342296e-04, -1.6814021e-03, ...,\n","                  -2.8856899e-04, -3.4555167e-05, -5.7278766e-04],\n","                 ...,\n","                 [-8.8731764e-04,  1.7460363e-04, -1.9392512e-05, ...,\n","                   1.7278090e-03,  6.6357438e-04,  3.7961287e-04],\n","                 [ 1.3298269e-03,  6.0328265e-04,  7.9002412e-04, ...,\n","                   4.0018596e-04, -1.6767619e-04, -3.2267891e-04],\n","                 [ 4.1007804e-04, -9.0255380e-05,  9.8878890e-04, ...,\n","                   5.1449670e-04, -3.9869166e-04,  1.0355078e-03]], dtype=float32),\n","          array([-6.6208740e-06,  1.6579761e-04,  7.7149329e-05, ...,\n","                 -1.5305448e-04, -1.5847391e-04,  1.3679601e-04], dtype=float32)),\n","         ((array([[ 7.4393797e-04,  7.2109926e-04, -7.4571476e-04, ...,\n","                    1.4219646e-03,  9.0123031e-05, -1.4654747e-03],\n","                  [ 2.5376831e-03, -4.8192174e-04, -2.1183370e-03, ...,\n","                   -1.0156085e-03, -5.7593279e-04, -7.9220037e-05],\n","                  [-1.1036985e-03,  1.0882003e-03, -2.7267449e-04, ...,\n","                    3.4653515e-04, -3.6296397e-04, -2.7548760e-04],\n","                  ...,\n","                  [ 4.0830808e-05,  5.0285819e-04,  7.5622555e-04, ...,\n","                   -8.5773550e-05, -1.4571253e-03, -7.6606448e-05],\n","                  [ 4.3986517e-04, -1.6245946e-03,  1.9941218e-03, ...,\n","                    1.0731750e-04,  1.7299330e-04,  3.2349312e-04],\n","                  [-8.9539553e-06,  3.5451498e-04, -1.2705188e-04, ...,\n","                    4.2292147e-04, -1.2856113e-03, -7.7409716e-04]], dtype=float32),\n","           array([ 1.0098539e-03, -3.0737370e-04,  5.8696376e-05, ...,\n","                   5.2702904e-04,  8.4068626e-04,  3.6080313e-04], dtype=float32)),),\n","         ((array([[ 6.7510392e-04,  1.2554860e-04,  9.2854095e-04, ...,\n","                   -2.6597886e-04,  2.5517176e-04,  1.9529798e-04],\n","                  [-1.2615547e-03, -1.3881843e-04,  9.9258439e-04, ...,\n","                    8.0496183e-04,  5.3649559e-04,  2.5984459e-04],\n","                  [ 1.4525790e-04,  1.9998405e-04, -1.1041548e-04, ...,\n","                    3.3759268e-06,  3.2450559e-04, -3.3069515e-04],\n","                  ...,\n","                  [ 9.8799069e-07, -4.7737136e-04,  7.0399656e-05, ...,\n","                   -4.7425533e-04,  1.6776590e-04,  1.3746745e-04],\n","                  [-3.1960467e-04, -4.5538411e-04,  1.9903401e-04, ...,\n","                   -2.0236934e-05,  4.6396762e-04, -4.8829429e-04],\n","                  [ 2.5265303e-04,  8.7623579e-05, -1.3480965e-05, ...,\n","                   -6.9342530e-04, -2.3998119e-04, -5.2886342e-05]], dtype=float32),\n","           array([ 8.1488710e-05,  1.4476389e-04,  5.1795889e-04, ...,\n","                   4.0061114e-04, -1.0126320e-04, -4.3189673e-05], dtype=float32)),)),\n","        (),\n","        (array([[ 3.9917216e-04, -5.0296600e-04,  1.9699392e-04, ...,\n","                  6.3325505e-04,  3.9124023e-04,  5.1898981e-04],\n","                [ 4.8749786e-04, -7.0589420e-05,  1.1171855e-04, ...,\n","                  7.3525094e-05, -5.4543074e-05,  3.9246332e-04],\n","                [-4.4532976e-04, -5.7994603e-04,  8.4190397e-05, ...,\n","                  2.8017655e-04,  7.9201236e-05,  5.7927461e-04],\n","                ...,\n","                [ 3.1043869e-04,  2.1387309e-04,  3.3291155e-05, ...,\n","                 -7.4079871e-04, -1.0298588e-03,  9.0624485e-04],\n","                [-2.3230848e-04, -1.2378621e-05,  7.9749487e-05, ...,\n","                  3.3096730e-04, -2.9105914e-04, -3.3117569e-04],\n","                [-6.0355978e-04, -1.3148585e-04,  5.0265662e-04, ...,\n","                 -9.5073390e-04,  1.6755878e-04, -5.3072326e-05]], dtype=float32),\n","         array([ 1.6123980e-04, -1.4621238e-04,  5.1728945e-05, ...,\n","                -3.9766976e-05, -1.5276916e-04,  1.0133709e-04], dtype=float32))), ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([27.710892, 26.12562 , 30.968655, ..., 36.47299 , 25.994028,\n","              35.38403 ], dtype=float32),\n","       array([-0.00040708, -0.00177415, -0.00146067, ...,  0.00065929,\n","              -0.00192971,  0.00196727], dtype=float32)),\n","      (array([[-1.3043759e-03, -1.7790920e-04,  5.5646480e-05, ...,\n","               -1.6827721e-04,  2.9022229e-04, -2.5395674e-04],\n","              [ 4.6134173e-04,  2.4408667e-04,  1.9190059e-04, ...,\n","               -9.4457384e-04, -4.7045751e-04,  5.8885186e-04],\n","              [ 1.0866391e-03, -3.2344458e-04, -2.4962708e-04, ...,\n","                3.8474277e-04, -6.8751920e-04,  8.7994100e-05],\n","              ...,\n","              [-1.2371629e-03, -5.7145389e-04,  1.2533883e-04, ...,\n","                4.5867436e-04,  1.4925557e-04, -3.5409426e-04],\n","              [ 6.4807519e-04,  2.8373368e-04,  2.1399686e-04, ...,\n","                6.2161777e-04, -1.0379614e-04,  2.1341087e-04],\n","              [ 3.4468336e-05, -2.7563989e-05,  7.2749925e-04, ...,\n","                5.6254509e-04,  4.3545285e-04, -2.6680841e-04]], dtype=float32),\n","       array([-0.00017006, -0.00025486, -0.00027613, ..., -0.00019427,\n","              -0.00020529, -0.00023063], dtype=float32)),\n","      ((), (), ()),\n","      (),\n","      (array([[-3.0325731e-04,  9.0084935e-04, -6.0674606e-04, ...,\n","                8.6357024e-05, -1.9538014e-04, -5.1961665e-04],\n","              [ 6.7488640e-04, -3.6897557e-04,  4.0472916e-04, ...,\n","                9.4491523e-04, -2.0013240e-04,  2.9562399e-04],\n","              [ 2.8856390e-04,  8.1856491e-04, -1.0262697e-04, ...,\n","                1.7339470e-04,  7.3503167e-04, -4.5893805e-05],\n","              ...,\n","              [-4.3178268e-04, -8.8659587e-04,  2.2125601e-04, ...,\n","                2.3206745e-05,  6.1015401e-04,  4.1297561e-04],\n","              [ 6.1883079e-04,  9.6077092e-05, -3.4040288e-04, ...,\n","               -4.3644399e-05,  3.0147191e-04, -5.0269492e-04],\n","              [-2.0641414e-04,  5.6721555e-04, -7.5222597e-05, ...,\n","                6.9334441e-05,  5.0178415e-04, -3.8402266e-04]], dtype=float32),\n","       array([ 4.0819552e-05, -3.2312266e-04,  1.8120308e-04, ...,\n","               1.4081175e-04, -4.8179234e-05, -4.3064745e-05], dtype=float32)),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([19.04798 , 20.078104, 21.36673 , ..., 19.276464, 19.51329 ,\n","              24.02911 ], dtype=float32),\n","       array([-1.8264142e-04,  3.3695583e-04, -4.0465326e-05, ...,\n","               1.4313638e-04,  4.6498002e-04,  1.7377081e-04], dtype=float32)),\n","      ((),\n","       ((),\n","        (((), (((array([[ 3.6010472e-04,  9.2087028e-04, -2.0830246e-05, ...,\n","                     -2.2231869e-04, -1.6758862e-04, -1.7619647e-04],\n","                    [-4.8282198e-04,  1.2845906e-03, -9.5843489e-04, ...,\n","                     -5.5589405e-04, -1.1122763e-03, -5.5668445e-04],\n","                    [ 6.6883641e-04,  4.6629473e-04, -7.6098467e-04, ...,\n","                     -9.5753669e-05,  4.9478095e-04,  9.3853468e-04],\n","                    ...,\n","                    [-1.4542885e-03, -1.0792821e-05,  1.4099224e-04, ...,\n","                      1.8259694e-05,  2.2292964e-04, -4.1351988e-04],\n","                    [ 6.6879508e-04, -4.6054143e-04,  8.2105619e-04, ...,\n","                     -7.0435728e-04,  6.6065431e-06, -3.8540675e-04],\n","                    [ 1.5417646e-03, -5.3607847e-04, -6.1273633e-04, ...,\n","                      1.4800460e-03,  2.4693236e-03, -1.0365077e-03]], dtype=float32),\n","             array([ 5.4594635e-05,  1.6149008e-04,  4.0367369e-05, ...,\n","                     6.2079023e-04,  1.8696285e-04, -1.7274776e-04], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[ 8.2947360e-04,  6.0554402e-04,  1.3333332e-04, ...,\n","                      7.5746147e-04,  3.4742706e-04, -5.0866901e-04],\n","                    [ 3.1374372e-04,  4.7216966e-04, -7.9428259e-04, ...,\n","                     -1.2523086e-04,  1.1470659e-03, -9.5631595e-04],\n","                    [-8.9870841e-04, -5.7019916e-04,  9.4758527e-04, ...,\n","                     -2.0877202e-04, -2.9314320e-05,  6.7141216e-04],\n","                    ...,\n","                    [-9.4727654e-04,  5.2437617e-04, -3.3521012e-04, ...,\n","                     -1.5441277e-04, -7.7100226e-04, -5.9744564e-04],\n","                    [ 8.9776295e-04, -9.4584521e-04,  3.6693097e-04, ...,\n","                      2.8946588e-04,  2.0598806e-04, -6.4851815e-04],\n","                    [ 7.0213187e-06, -5.3501234e-04,  2.8118264e-04, ...,\n","                     -2.5776951e-04,  1.5052332e-03, -4.7607301e-04]], dtype=float32),\n","             array([ 1.8742301e-04, -1.3134445e-04,  7.7326067e-06, ...,\n","                     6.0479727e-04, -2.6051910e-04,  3.7350276e-04], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[-6.2084757e-04,  2.9438350e-04, -3.6325218e-04, ...,\n","                     -3.7355023e-04,  4.3579508e-04,  1.1944018e-03],\n","                    [-9.0757658e-06,  3.6502961e-04, -6.6654495e-04, ...,\n","                      7.9885498e-04,  7.6183671e-05, -5.5072788e-04],\n","                    [ 7.4112357e-04, -1.1528708e-04, -4.5243170e-04, ...,\n","                     -1.0911763e-03, -5.3643034e-04, -3.8295766e-05],\n","                    ...,\n","                    [-3.0499999e-04, -5.1435316e-04, -2.4503851e-04, ...,\n","                     -4.0117683e-04,  1.4186653e-04,  4.5977620e-05],\n","                    [ 1.2393923e-03, -1.4760972e-04, -1.2060060e-03, ...,\n","                      8.2538199e-06, -6.4167347e-05, -7.0968556e-04],\n","                    [-2.1729382e-05,  2.7222157e-04, -1.0066603e-03, ...,\n","                      1.1004242e-03, -9.6222403e-04,  6.6643429e-04]], dtype=float32),\n","             array([-1.8956154e-05,  7.6492441e-05,  9.4052601e-05, ...,\n","                     1.2405234e-04,  1.7842067e-04,  7.1825139e-05], dtype=float32)),\n","            ((), (), ())))),\n","         (),\n","         ((), (), ()),\n","         (array([[ 1.12408576e-04,  5.97732025e-04, -1.19317738e-04, ...,\n","                   4.95012093e-04, -6.12319127e-05, -2.89442745e-04],\n","                 [ 4.89072583e-04,  3.92282818e-04,  1.48564968e-05, ...,\n","                  -1.63214252e-04, -6.98251242e-05,  1.40928067e-04],\n","                 [ 1.90925071e-04, -7.32595799e-05, -1.20204408e-03, ...,\n","                   1.76549191e-04, -7.12519322e-05, -1.51819811e-04],\n","                 ...,\n","                 [ 3.47063062e-04,  4.27551888e-04,  2.06731173e-04, ...,\n","                   1.69146253e-04, -2.19462949e-04,  7.97480170e-04],\n","                 [ 2.59161752e-04,  1.09278853e-03, -1.75272304e-04, ...,\n","                   6.53610157e-04, -1.30411889e-03, -3.88262299e-04],\n","                 [ 8.27275100e-04,  5.57283231e-04, -1.90140690e-05, ...,\n","                   6.79576828e-04,  7.25286445e-05,  2.73739744e-04]], dtype=float32),\n","          array([ 4.4025204e-05, -2.0857059e-04,  1.2490437e-04, ...,\n","                 -3.5981509e-05, -9.2135473e-05,  1.6803671e-04], dtype=float32))),\n","        ()),\n","       ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([27.589556, 23.701832, 29.036041, ..., 26.093735, 25.85335 ,\n","              28.782913], dtype=float32),\n","       array([-0.00050414, -0.00014547,  0.00045822, ...,  0.00019999,\n","              -0.00010039, -0.00044606], dtype=float32)),\n","      (),\n","      ((), (((array([[-7.7539729e-04, -5.8742095e-04,  4.3319087e-04, ...,\n","                   4.6159334e-05,  5.0548097e-04,  4.2778638e-04],\n","                 [-1.1671141e-03, -2.4256085e-04, -6.1501755e-04, ...,\n","                  -4.5737735e-04, -4.5597611e-04, -1.1788093e-03],\n","                 [ 3.8539775e-04, -2.0298407e-04,  3.7861214e-04, ...,\n","                   2.5512130e-04, -7.9322362e-04,  3.6517764e-04],\n","                 ...,\n","                 [-2.4389361e-04,  7.6851301e-04, -5.0681649e-04, ...,\n","                   1.2031368e-03,  6.2267226e-04, -1.5304753e-04],\n","                 [-6.0807064e-04, -1.2352382e-04, -1.1467651e-03, ...,\n","                  -7.3625805e-04,  4.8502698e-05,  2.9808781e-04],\n","                 [-8.4430515e-04,  8.9628338e-05,  1.4257561e-03, ...,\n","                   4.8383445e-04,  5.5990776e-04,  1.8411016e-04]], dtype=float32),\n","          array([ 6.73655813e-05, -8.89110161e-05,  1.33064741e-05, ...,\n","                 -1.09979665e-04, -4.57331946e-04,  1.69093018e-05], dtype=float32)),\n","         ((array([[ 7.3721906e-04,  9.7429787e-04,  7.7946135e-04, ...,\n","                   -9.4295014e-05, -1.9704069e-04, -1.2278965e-03],\n","                  [ 3.2161045e-04, -1.2436578e-03, -4.7227580e-04, ...,\n","                   -8.7843864e-04, -1.8094951e-06,  5.1287486e-04],\n","                  [ 1.1111080e-03,  8.3910528e-04, -5.2709639e-04, ...,\n","                   -5.9197325e-04,  5.0049566e-04,  1.7485529e-03],\n","                  ...,\n","                  [ 8.9801673e-04,  5.3159788e-04, -5.2576797e-04, ...,\n","                    2.1112410e-04,  2.8968227e-04, -1.3745498e-03],\n","                  [-1.3354189e-03,  5.7428417e-04, -1.7677282e-04, ...,\n","                    7.7160302e-04,  2.2173834e-05, -9.7149267e-04],\n","                  [-1.0097915e-03, -6.0993632e-05,  1.0128116e-03, ...,\n","                   -1.6424150e-03, -1.5145292e-03, -9.2446024e-04]], dtype=float32),\n","           array([ 0.00046772, -0.00015372, -0.00050895, ..., -0.00039174,\n","                  -0.00024612,  0.0005687 ], dtype=float32)),),\n","         ((array([[-1.5732647e-04,  1.9570859e-04,  5.3472584e-04, ...,\n","                    1.5731435e-04,  5.2068621e-04,  5.0832226e-04],\n","                  [-6.5022969e-04,  1.2052072e-03,  2.5600066e-06, ...,\n","                   -1.4521914e-04,  7.8667910e-04,  1.5292806e-04],\n","                  [ 2.0848861e-04, -1.2043928e-04,  2.4985697e-04, ...,\n","                   -5.0362851e-04, -5.6959834e-04, -3.4715957e-04],\n","                  ...,\n","                  [ 6.0039380e-04, -1.3831999e-04,  4.2565085e-04, ...,\n","                    6.7795845e-05, -8.8637939e-04, -1.0018161e-04],\n","                  [ 3.0762417e-04,  5.3542835e-04,  1.1036264e-03, ...,\n","                   -4.8523230e-05,  3.0267925e-04, -5.1660417e-04],\n","                  [ 3.7181814e-04, -6.3725776e-04, -4.0213465e-05, ...,\n","                   -3.7710424e-04, -2.3810616e-04, -1.3759814e-04]], dtype=float32),\n","           array([ 6.32559168e-05,  2.17419947e-04,  2.57075473e-04, ...,\n","                  -4.96054236e-06,  1.14224866e-04, -4.40466283e-05], dtype=float32)),)),\n","        (),\n","        (array([[ 1.2812532e-04,  3.7940327e-04, -8.3720843e-05, ...,\n","                  1.6292417e-04,  5.3579203e-04,  2.0588367e-04],\n","                [-6.5250183e-04, -1.0296281e-03, -3.2464083e-04, ...,\n","                 -4.8715924e-04,  9.1673603e-04, -8.3357369e-04],\n","                [ 1.1494060e-03, -2.7775101e-04,  2.7043710e-04, ...,\n","                 -6.3358026e-04,  7.9351111e-04,  7.0637983e-04],\n","                ...,\n","                [-3.9354505e-04,  6.6409513e-05, -3.2691611e-04, ...,\n","                  2.8968658e-05, -3.4568753e-04,  1.8914837e-04],\n","                [-3.5601656e-04,  7.5230905e-04, -4.2020451e-04, ...,\n","                 -4.9785658e-04, -3.6123802e-04,  1.3626173e-04],\n","                [-6.5304973e-04, -1.7528915e-04, -1.9884703e-04, ...,\n","                  4.2483654e-05, -9.0664464e-05,  6.8377281e-05]], dtype=float32),\n","         array([ 6.89869485e-05, -1.19526616e-04,  2.21117327e-04, ...,\n","                 2.22060789e-05, -1.38805684e-04,  9.47177614e-05], dtype=float32))), ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([29.752275, 27.545883, 29.982363, ..., 33.795074, 28.028606,\n","              39.10248 ], dtype=float32),\n","       array([-1.0667844e-03,  2.0834578e-03,  3.6616922e-05, ...,\n","              -3.4254619e-03, -3.0797147e-03,  1.0013189e-03], dtype=float32)),\n","      (array([[-2.27076322e-04,  1.20826800e-04, -1.30966815e-04, ...,\n","               -1.10489345e-04,  3.12164921e-04, -3.32416501e-04],\n","              [-5.86470822e-04, -1.91527623e-04, -4.91580286e-04, ...,\n","               -1.82450880e-04,  1.86832083e-04,  5.28978017e-05],\n","              [-9.67565356e-05, -2.40298657e-04, -4.53582907e-05, ...,\n","                6.58296864e-04,  1.99634655e-04,  2.41860922e-04],\n","              ...,\n","              [ 1.17946598e-04,  7.07265397e-04,  6.57863508e-04, ...,\n","               -1.66792961e-04, -3.66251101e-04, -2.50053155e-04],\n","              [-3.17123544e-04,  6.64376537e-04, -1.44096877e-04, ...,\n","               -1.16184747e-04, -1.21887086e-03, -4.78190370e-04],\n","              [ 5.86854934e-04,  1.92289779e-04, -2.49661738e-04, ...,\n","                5.99353109e-04,  8.25965835e-04,  3.22049193e-04]], dtype=float32),\n","       array([-0.00033306, -0.00026527, -0.00020221, ..., -0.00028069,\n","              -0.00029134, -0.00021733], dtype=float32)),\n","      ((), (), ()),\n","      (),\n","      (array([[-5.1405281e-04, -5.5886427e-04, -1.0630088e-04, ...,\n","                4.0257251e-04, -4.3637230e-04,  6.0328055e-04],\n","              [-4.8121881e-05, -3.1568465e-04, -2.4826278e-04, ...,\n","               -6.2883354e-04,  1.5627536e-04,  7.2795508e-04],\n","              [ 3.2441528e-04, -2.2408024e-04,  2.1705774e-04, ...,\n","                9.8179886e-04,  1.4929505e-04, -5.4126245e-04],\n","              ...,\n","              [-5.1092234e-04,  3.1988876e-04,  4.5957547e-04, ...,\n","                7.3456095e-04, -4.2980787e-04, -5.4370676e-04],\n","              [-6.6416146e-04,  1.8510839e-04,  2.5869795e-04, ...,\n","                5.0931086e-04, -2.8166719e-04,  4.6520980e-04],\n","              [-1.9260134e-04, -6.3677097e-04,  1.9456517e-04, ...,\n","               -2.2721342e-04, -6.7265652e-04,  5.2278308e-04]], dtype=float32),\n","       array([-1.7535579e-04, -2.6856700e-04,  3.0355281e-04, ...,\n","               2.0664462e-04, -3.3068918e-05, -6.9814909e-05], dtype=float32)),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([7.464822 , 8.624882 , 6.0506716, ..., 6.5541067, 7.722385 ,\n","              6.5531573], dtype=float32),\n","       array([-1.5464889e-06,  8.4128282e-05,  3.8028316e-04, ...,\n","               1.8282833e-07,  1.3350374e-04,  1.9292452e-04], dtype=float32)),\n","      ((),\n","       ((),\n","        (((), (((array([[ 6.9233205e-04,  8.7018288e-04, -7.4935338e-04, ...,\n","                      4.5910149e-04,  3.6925400e-04, -5.1886449e-04],\n","                    [-1.4749052e-03, -8.4540673e-04,  2.5374084e-03, ...,\n","                      1.0529242e-03, -6.4389933e-05, -9.2016615e-04],\n","                    [-1.7194348e-04, -6.2564976e-04,  6.4950186e-04, ...,\n","                      3.9346221e-05,  9.3276874e-04, -1.1453629e-04],\n","                    ...,\n","                    [ 2.6843877e-04,  7.9030244e-05, -1.1376322e-03, ...,\n","                      8.3741779e-04,  7.3966097e-05, -8.6823560e-04],\n","                    [ 1.2297912e-04,  1.0923469e-03,  6.7530852e-04, ...,\n","                     -1.0067791e-03, -1.0264020e-03,  4.8175442e-04],\n","                    [ 1.6233858e-04, -1.5873736e-03,  1.7076426e-03, ...,\n","                     -3.1689811e-04,  6.4890296e-04,  7.7084603e-04]], dtype=float32),\n","             array([ 1.12488604e-04, -2.83380912e-04, -5.92886281e-05, ...,\n","                    -1.09753521e-04, -1.50217747e-04,  1.84272969e-04], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[ 0.00017915,  0.00054876, -0.00148472, ..., -0.00191991,\n","                      0.00028687,  0.00226707],\n","                    [ 0.0019887 , -0.00259463,  0.00241257, ..., -0.00108891,\n","                     -0.00049202,  0.00076539],\n","                    [ 0.00083976, -0.0013735 ,  0.00130427, ...,  0.00064281,\n","                      0.00029446, -0.00059514],\n","                    ...,\n","                    [-0.00026698,  0.00068535, -0.00023223, ...,  0.00023919,\n","                     -0.00049501,  0.00026959],\n","                    [-0.00055808,  0.00256885, -0.00161576, ...,  0.00113494,\n","                      0.00071377, -0.00108078],\n","                    [-0.00075156, -0.00026378,  0.0006259 , ...,  0.00024291,\n","                     -0.00035195, -0.00068614]], dtype=float32),\n","             array([-4.6199720e-04, -2.6904634e-04, -2.4120232e-04, ...,\n","                     5.6761164e-05,  2.9859619e-04, -6.5895176e-05], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[ 1.77109905e-04,  2.64258124e-04,  4.55074798e-04, ...,\n","                      2.99797510e-04, -2.29898607e-04,  3.30039155e-04],\n","                    [-7.16564464e-05, -5.38453402e-04,  7.43359677e-04, ...,\n","                     -3.53926240e-04, -5.66690520e-04,  5.41656045e-04],\n","                    [-4.96459368e-04, -6.18339807e-04,  5.47646923e-05, ...,\n","                      8.07660763e-05, -2.98508501e-04, -1.45878206e-04],\n","                    ...,\n","                    [-5.23582858e-04, -3.33722215e-04, -3.75487318e-04, ...,\n","                      1.13595110e-04,  2.02124153e-04,  1.12640075e-04],\n","                    [ 6.14479883e-04, -5.41831068e-05,  4.49403509e-04, ...,\n","                      3.50454320e-05,  1.29559616e-04, -4.41576209e-04],\n","                    [ 8.09310877e-04,  7.64188517e-05,  7.40562042e-04, ...,\n","                      1.59868156e-04, -4.46963211e-04, -1.63224031e-04]], dtype=float32),\n","             array([ 9.9415010e-05,  1.2790141e-04,  3.7990147e-04, ...,\n","                    -1.7073300e-05, -1.5485659e-05, -1.4334895e-05], dtype=float32)),\n","            ((), (), ())))),\n","         (),\n","         ((), (), ()),\n","         (array([[-3.3713077e-04,  4.9304077e-04,  3.1833893e-05, ...,\n","                  -3.3303979e-04, -3.3409472e-04, -3.4498001e-04],\n","                 [ 2.5460508e-04, -2.1338173e-04, -2.5358726e-04, ...,\n","                   3.7557809e-04,  4.4838485e-04, -1.5286639e-04],\n","                 [ 8.6696330e-04,  3.7816371e-04,  6.3227437e-04, ...,\n","                   2.1695405e-04, -8.3598407e-04, -2.6168427e-04],\n","                 ...,\n","                 [-4.3695363e-05,  3.7777867e-05,  6.4898236e-04, ...,\n","                   5.1685376e-04, -2.6554786e-04, -1.1451123e-04],\n","                 [ 5.6271598e-05,  9.8023620e-06,  4.6138553e-04, ...,\n","                  -1.9689668e-04,  2.3118651e-05,  3.3558175e-04],\n","                 [-5.7511742e-04, -5.1461894e-04, -2.8365946e-04, ...,\n","                  -3.9686236e-04,  1.0337373e-05,  4.7793828e-06]], dtype=float32),\n","          array([ 2.9710509e-05, -1.7216493e-04,  1.9841531e-05, ...,\n","                  9.9785400e-05, -7.7276592e-05,  2.1051220e-04], dtype=float32))),\n","        ()),\n","       ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([32.983616, 23.116045, 27.992409, ..., 28.093359, 23.628382,\n","              29.797533], dtype=float32),\n","       array([-2.0265250e-04, -3.7406047e-04, -8.6957873e-05, ...,\n","               3.9543177e-04, -3.9789907e-04, -1.0733848e-04], dtype=float32)),\n","      (),\n","      ((), (((array([[-1.5214224e-04, -3.4591052e-04, -7.3691685e-05, ...,\n","                  -3.6239327e-04,  1.1267124e-03,  2.4454723e-04],\n","                 [ 1.2596807e-04,  9.0011122e-04, -4.8422933e-04, ...,\n","                   1.3687302e-05,  1.5135146e-04, -7.0751697e-04],\n","                 [ 9.8083180e-04, -1.6279820e-04,  2.4963266e-04, ...,\n","                  -1.5419035e-04,  1.6663241e-04,  6.9173786e-04],\n","                 ...,\n","                 [ 2.9130903e-04, -1.0282055e-04,  6.6496973e-04, ...,\n","                   2.6064733e-04,  2.0592564e-03,  8.7975257e-04],\n","                 [ 6.1888288e-04,  1.7990952e-04, -4.5736003e-04, ...,\n","                   2.3611620e-04, -8.3330908e-04, -5.6554069e-05],\n","                 [-5.5525568e-04,  5.2804866e-05, -1.9562287e-04, ...,\n","                  -1.7311835e-03, -3.3579685e-04,  8.5196097e-04]], dtype=float32),\n","          array([-8.2365354e-05, -6.3164598e-06,  5.0827028e-04, ...,\n","                  1.6769100e-04, -2.2698744e-04,  1.8175927e-04], dtype=float32)),\n","         ((array([[ 1.2050428e-03,  1.5666681e-04, -1.3660904e-03, ...,\n","                   -3.4422163e-05, -1.1119717e-03,  1.9796255e-04],\n","                  [ 4.3317303e-04,  3.0687434e-04, -3.1355390e-04, ...,\n","                   -6.5480440e-04, -2.9218505e-04, -1.1140192e-03],\n","                  [-1.4695117e-06, -2.2644480e-04,  4.0673331e-04, ...,\n","                    2.6596227e-04, -1.0557702e-04, -1.9442412e-04],\n","                  ...,\n","                  [-1.0782661e-03, -1.1412369e-04,  1.1586620e-04, ...,\n","                    2.6872658e-04, -6.6258642e-04, -1.3763668e-03],\n","                  [ 4.8703482e-04, -1.7324882e-04, -4.3247059e-05, ...,\n","                    5.8507768e-04, -8.6665968e-04,  1.2490185e-03],\n","                  [ 6.2134891e-04, -1.4701251e-04,  8.2691648e-04, ...,\n","                    6.8997149e-04, -1.4069782e-03, -2.4678765e-04]], dtype=float32),\n","           array([-1.7513359e-04, -5.0619121e-05, -1.6728892e-04, ...,\n","                  -1.8353190e-04,  1.5833913e-04, -1.5163490e-04], dtype=float32)),),\n","         ((array([[ 3.5116912e-04, -2.4633951e-04, -4.1339942e-04, ...,\n","                   -8.1207996e-05, -5.6523142e-05,  3.0814961e-04],\n","                  [ 7.5781431e-06, -5.2976777e-04, -6.2921294e-04, ...,\n","                   -9.0135518e-04,  3.0740243e-04,  8.6093984e-05],\n","                  [-2.1887257e-05,  2.5112813e-05, -1.2493301e-04, ...,\n","                    1.3871313e-04, -1.0272875e-04,  9.5418356e-05],\n","                  ...,\n","                  [ 5.9851527e-04, -6.0335879e-04, -3.6743350e-04, ...,\n","                   -8.2830089e-04,  4.3310341e-04,  3.6641673e-04],\n","                  [ 5.0648727e-04,  6.0449494e-04,  2.8506448e-04, ...,\n","                   -1.7189564e-05, -5.5930990e-04,  2.0906515e-04],\n","                  [-3.1179693e-04, -5.6581304e-04,  3.1488959e-04, ...,\n","                    3.2698031e-04,  3.1401293e-04,  3.0611546e-04]], dtype=float32),\n","           array([ 7.5799517e-06, -1.3069410e-04,  9.0992704e-05, ...,\n","                   6.0800900e-04,  6.9105619e-04,  9.6983335e-04], dtype=float32)),)),\n","        (),\n","        (array([[ 6.01515174e-04,  4.64601035e-04,  2.58323835e-05, ...,\n","                  6.22327789e-04, -1.11058944e-04, -3.05324706e-04],\n","                [ 5.62672387e-04,  7.39778683e-04,  6.43129024e-05, ...,\n","                  5.23757655e-04, -2.78443942e-04,  8.36059160e-04],\n","                [ 1.57524628e-04,  2.61111825e-04, -3.44780041e-04, ...,\n","                 -2.20612375e-04,  7.53933738e-04, -4.28425061e-04],\n","                ...,\n","                [ 1.98073918e-04, -3.03321838e-04,  4.05029539e-04, ...,\n","                 -2.82543915e-04,  4.79473601e-05, -7.17827585e-04],\n","                [ 6.16310339e-04, -3.03807290e-04,  2.60822388e-04, ...,\n","                 -8.73113982e-04, -1.57233648e-04,  2.30891688e-04],\n","                [ 4.90726612e-04,  6.38225989e-04,  3.93784139e-04, ...,\n","                  9.85301740e-05,  3.78726749e-04, -4.81044553e-04]], dtype=float32),\n","         array([ 1.9104987e-04, -9.1049973e-05,  1.5905402e-04, ...,\n","                 3.0058087e-05,  6.0540286e-05,  3.4530109e-04], dtype=float32))), ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([27.615122, 26.078981, 30.52405 , ..., 32.489735, 29.551847,\n","              35.766136], dtype=float32),\n","       array([-0.00194025,  0.00179261, -0.00087198, ..., -0.00316836,\n","               0.00105329,  0.00452777], dtype=float32)),\n","      (array([[ 5.6696311e-04, -1.9348484e-04, -5.5132713e-04, ...,\n","                2.3293316e-05, -5.2967016e-04, -5.0747796e-04],\n","              [ 2.3308481e-04, -2.3342248e-04, -1.0964255e-04, ...,\n","                4.3530038e-04,  4.7340014e-04, -8.3439169e-04],\n","              [ 4.8238380e-04, -3.2481932e-04, -3.1239295e-04, ...,\n","               -7.6285709e-04, -6.2411756e-04, -1.7298011e-06],\n","              ...,\n","              [ 1.8937860e-04,  4.9852533e-04, -3.4654394e-04, ...,\n","               -3.9495187e-04,  4.3786340e-04,  6.4942215e-05],\n","              [-1.5452581e-04,  3.1933768e-04,  8.6265733e-05, ...,\n","               -3.7432424e-04, -6.3204922e-04, -3.8573134e-04],\n","              [ 4.6715376e-04, -3.3290300e-04, -1.4276478e-04, ...,\n","               -3.7995618e-04,  4.0405526e-04,  2.7339775e-04]], dtype=float32),\n","       array([-4.5991917e-05, -2.3109744e-04, -2.1729936e-04, ...,\n","              -2.7812916e-04, -1.3777032e-04, -2.3623457e-04], dtype=float32)),\n","      ((), (), ()),\n","      (),\n","      (array([[-2.66631960e-05,  1.04967876e-04, -6.31503412e-04, ...,\n","               -3.61133076e-04,  3.94006842e-04, -2.30790669e-04],\n","              [ 1.03126036e-03,  1.33821537e-04,  9.05729714e-04, ...,\n","               -3.75407457e-04, -2.56853207e-04,  5.73388010e-04],\n","              [ 1.39090087e-04, -5.85161382e-04, -6.91839814e-05, ...,\n","                1.45021535e-04, -3.80372017e-04,  1.86760561e-04],\n","              ...,\n","              [ 4.20494907e-04, -9.09068767e-05, -3.32629075e-04, ...,\n","               -2.23558673e-04,  5.07426972e-04,  7.78009882e-04],\n","              [ 5.07526158e-04, -2.21028076e-05,  4.26888815e-04, ...,\n","                5.27065073e-04,  4.08268388e-04,  1.53284218e-05],\n","              [ 1.21821195e-05,  1.01897704e-05, -1.89182410e-05, ...,\n","               -8.56767118e-04,  4.92826766e-05,  9.41755308e-04]], dtype=float32),\n","       array([-2.4322799e-04, -7.7921359e-05,  4.0194427e-04, ...,\n","               2.0153956e-04, -2.2428462e-04, -1.8175803e-04], dtype=float32)),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([4.855751 , 6.989046 , 4.7327175, ..., 4.012898 , 6.2368827,\n","              4.382704 ], dtype=float32),\n","       array([-1.99295828e-05, -2.54071812e-04,  7.06891238e-04, ...,\n","               1.86548728e-04, -2.09612495e-04,  1.10228306e-04], dtype=float32)),\n","      ((),\n","       ((),\n","        (((), (((array([[ 1.5145150e-04,  2.9722569e-04,  7.8022125e-04, ...,\n","                     -1.6719057e-03,  1.5078310e-03, -2.3530319e-03],\n","                    [ 3.9493662e-04, -7.0981732e-05,  3.4332232e-04, ...,\n","                      1.7339257e-03, -2.9195063e-03,  2.0068933e-03],\n","                    [ 4.0458518e-04, -6.8559777e-04,  1.5814052e-03, ...,\n","                     -2.5653718e-03,  1.4648363e-03, -2.2346142e-03],\n","                    ...,\n","                    [-1.4300499e-03,  1.2332799e-03, -4.3616380e-04, ...,\n","                      6.8912911e-04, -5.3848559e-04,  8.7953144e-04],\n","                    [ 1.7592874e-03, -2.4138817e-03,  6.4958651e-05, ...,\n","                     -2.4534760e-05,  8.2347012e-04, -3.5783715e-04],\n","                    [-1.9525838e-03,  1.9181245e-03, -1.8534017e-03, ...,\n","                     -1.0833311e-03,  5.1880907e-04, -1.0062875e-03]], dtype=float32),\n","             array([-1.1637109e-04, -8.7675769e-05, -7.3491054e-04, ...,\n","                    -4.4897408e-04,  5.9189356e-04, -3.5705376e-05], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[-5.74302510e-04, -3.07292154e-04, -2.48441077e-03, ...,\n","                     -2.64492072e-03,  1.63651607e-03, -6.26683584e-04],\n","                    [-1.00046932e-03,  6.97803160e-04, -1.00310004e-04, ...,\n","                     -3.39365681e-03,  1.45498663e-03, -4.99891525e-04],\n","                    [ 5.15878201e-04, -7.30862143e-04, -1.03286700e-03, ...,\n","                     -1.09100994e-03,  3.59706930e-04,  2.93263729e-04],\n","                    ...,\n","                    [-2.22808449e-04,  7.26943981e-05,  3.63279862e-04, ...,\n","                     -8.43957969e-05, -3.30810435e-05, -9.56408505e-04],\n","                    [ 1.32671511e-03, -1.44449202e-03, -2.38661669e-04, ...,\n","                      1.97059591e-03, -1.52578868e-03,  1.68249069e-03],\n","                    [ 1.37812330e-03, -4.57306771e-04,  1.16964139e-03, ...,\n","                      1.53411087e-03, -2.24240997e-04,  9.02779517e-04]], dtype=float32),\n","             array([1.0128311e-04, 4.6686189e-05, 1.7126492e-04, ..., 4.0105420e-05,\n","                    5.4495723e-04, 1.8566263e-04], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[ 1.05834726e-04,  1.89823622e-04, -3.80404876e-04, ...,\n","                     -2.45011586e-04, -4.77627036e-04, -1.44505257e-05],\n","                    [-5.17290100e-05,  2.73891725e-04, -5.34046208e-04, ...,\n","                      1.17040065e-03,  4.27282532e-04, -6.84081926e-04],\n","                    [-4.35522641e-04,  8.45388058e-06, -1.38231655e-04, ...,\n","                      1.32375717e-04,  1.17241078e-04, -1.80846837e-05],\n","                    ...,\n","                    [ 9.69535729e-04, -6.38123835e-04,  1.12375028e-05, ...,\n","                      1.56070455e-05,  1.55976333e-04,  2.80585577e-04],\n","                    [-7.56775786e-04,  6.50818052e-04,  7.07276515e-04, ...,\n","                      8.16371175e-05, -4.93334606e-04,  7.08353764e-04],\n","                    [-8.20520858e-04,  6.68078646e-05,  2.29138910e-04, ...,\n","                     -1.51304048e-04,  9.00625528e-05, -2.76426028e-04]], dtype=float32),\n","             array([ 1.0549296e-04,  3.2083251e-05, -2.8118124e-05, ...,\n","                     8.7655804e-05, -5.6867739e-05,  2.9940893e-05], dtype=float32)),\n","            ((), (), ())))),\n","         (),\n","         ((), (), ()),\n","         (array([[ 8.8390842e-04,  1.6134461e-04, -1.9523801e-04, ...,\n","                   6.6623659e-05, -1.7275226e-05,  2.5753974e-05],\n","                 [-5.4493611e-04, -5.7377090e-04, -2.1276693e-04, ...,\n","                   3.4380841e-04, -8.4533286e-04, -1.0730352e-05],\n","                 [ 8.7199494e-04,  1.1685930e-04, -2.7755124e-04, ...,\n","                  -1.6924796e-04, -4.0553710e-05,  3.7013699e-05],\n","                 ...,\n","                 [-5.1796145e-04, -3.0999283e-05, -1.8597856e-04, ...,\n","                   9.9232659e-07,  2.5161859e-04, -3.3019020e-05],\n","                 [ 4.0314066e-05,  6.7132566e-04, -3.8169394e-04, ...,\n","                   2.3004669e-04,  2.1434335e-04,  2.4886403e-04],\n","                 [ 2.9081662e-04, -1.9430053e-04,  6.0523249e-04, ...,\n","                  -3.8382213e-04,  1.7466937e-05, -6.4967258e-05]], dtype=float32),\n","          array([-1.6790720e-04,  8.6666278e-06,  5.6424287e-05, ...,\n","                  8.5430409e-05,  1.0531673e-05,  1.2336532e-04], dtype=float32))),\n","        ()),\n","       ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([30.662365, 26.403925, 26.579105, ..., 27.908953, 27.801018,\n","              26.150112], dtype=float32),\n","       array([-1.6015703e-04, -2.1956467e-04,  2.0429314e-04, ...,\n","               4.1153343e-04, -4.7519748e-04,  9.2773669e-05], dtype=float32)),\n","      (),\n","      ((), (((array([[ 1.4485412e-04,  1.7418525e-04, -1.2707158e-03, ...,\n","                   8.5957791e-04, -9.2425017e-04,  7.4577722e-04],\n","                 [-1.7107600e-04, -2.2835561e-04,  1.8639660e-04, ...,\n","                   1.0288818e-03,  7.8010239e-04, -3.1825883e-05],\n","                 [ 3.3249025e-04, -6.8760046e-04, -4.5525446e-04, ...,\n","                  -6.8438204e-04, -7.3356030e-04, -2.8497542e-04],\n","                 ...,\n","                 [-1.9546741e-04,  8.9571215e-05,  4.4941541e-04, ...,\n","                   7.7435223e-04, -2.9358795e-04, -4.0220711e-04],\n","                 [ 1.4873613e-03, -1.1542060e-03,  1.4392087e-03, ...,\n","                  -1.5877918e-05, -1.0029539e-05, -6.3591514e-04],\n","                 [-4.1896501e-04,  1.0447039e-03, -3.0632020e-04, ...,\n","                   3.9649912e-04,  5.8484497e-04,  2.5573446e-04]], dtype=float32),\n","          array([-3.3729619e-04, -2.3419077e-04,  1.8702693e-06, ...,\n","                 -5.0661899e-04,  1.9382359e-04, -7.5824995e-05], dtype=float32)),\n","         ((array([[ 5.8338325e-04,  8.5761468e-04,  4.3625908e-04, ...,\n","                    7.5149239e-04, -8.1684196e-04,  1.6839581e-03],\n","                  [-5.4987904e-04, -3.9409912e-05, -1.6243574e-04, ...,\n","                    4.0612111e-04, -3.3947936e-04,  7.1923935e-04],\n","                  [ 7.7410194e-04,  8.2182574e-05, -3.5489959e-04, ...,\n","                    1.4319507e-04, -7.5303437e-04, -3.5683133e-04],\n","                  ...,\n","                  [-4.2840422e-04, -6.9878681e-04, -1.0322904e-03, ...,\n","                    1.1141074e-03,  8.5411721e-04, -4.2964073e-04],\n","                  [-2.1600074e-04, -7.4265397e-04,  2.5637713e-03, ...,\n","                   -5.2877859e-04,  4.6443392e-04, -1.2105559e-03],\n","                  [-7.4833346e-04, -2.1043776e-03,  5.1165687e-05, ...,\n","                   -7.1253604e-04, -4.7327296e-05,  9.3354436e-04]], dtype=float32),\n","           array([ 6.0449102e-06, -1.5009668e-04, -7.7272431e-05, ...,\n","                  -1.3362558e-04, -8.6274140e-06,  7.0275251e-05], dtype=float32)),),\n","         ((array([[ 1.54884299e-04, -4.70323634e-04, -2.19079884e-04, ...,\n","                    6.94571470e-04,  1.55356247e-05, -1.53227928e-04],\n","                  [-1.98077556e-04, -7.33187306e-04,  5.97829348e-04, ...,\n","                   -1.68417901e-04, -8.35360435e-04, -5.03884570e-04],\n","                  [ 1.19385568e-05,  5.10720092e-05, -2.77150568e-04, ...,\n","                    2.33308470e-04, -1.40745702e-04, -5.14675703e-05],\n","                  ...,\n","                  [-1.49054278e-03,  4.71721462e-04,  6.43087042e-05, ...,\n","                    2.61119887e-04,  5.01459872e-04,  6.02888933e-04],\n","                  [ 7.74771383e-04,  1.44730380e-04, -5.79364365e-04, ...,\n","                    3.53923882e-04, -7.80505012e-04,  3.70974431e-06],\n","                  [ 2.28917015e-05, -6.77999284e-04,  1.16241405e-04, ...,\n","                    3.44372034e-04,  1.30655157e-04, -2.64185743e-04]], dtype=float32),\n","           array([ 0.00040237,  0.00056824, -0.00068751, ...,  0.00019244,\n","                   0.00040021,  0.00018913], dtype=float32)),)),\n","        (),\n","        (array([[ 2.6772477e-04,  6.0702040e-04,  2.5187040e-04, ...,\n","                  6.5034034e-04,  3.4297965e-04, -1.3800486e-04],\n","                [-5.0681795e-04,  5.9558428e-04,  7.5222844e-05, ...,\n","                 -3.3979915e-04, -3.4837413e-04,  2.9196549e-04],\n","                [ 8.0919988e-04,  3.8601653e-04,  3.1798013e-04, ...,\n","                  6.0521485e-04,  6.2834017e-04, -3.9738472e-04],\n","                ...,\n","                [ 3.0434836e-04, -8.7165000e-04,  1.5342720e-04, ...,\n","                  2.8413167e-04,  3.2655304e-04,  3.6846948e-04],\n","                [-3.1421205e-05,  1.3106583e-04, -1.2436017e-04, ...,\n","                  1.1926603e-05,  2.7038774e-04, -7.3099200e-04],\n","                [ 4.0909686e-04, -9.9063886e-04, -3.9278981e-04, ...,\n","                  5.8779761e-04,  1.5066817e-04, -6.8860437e-04]], dtype=float32),\n","         array([ 1.15930896e-04, -1.48469946e-04,  2.75138387e-04, ...,\n","                 1.24996677e-05,  2.05701544e-05,  7.13871996e-05], dtype=float32))), ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([30.850067, 25.940405, 26.610691, ..., 30.938204, 27.196863,\n","              28.740248], dtype=float32),\n","       array([-0.00115797,  0.00174931, -0.00140941, ..., -0.0016462 ,\n","               0.0011183 ,  0.00234788], dtype=float32)),\n","      (array([[-5.0629175e-04,  2.1610307e-04, -5.2092056e-04, ...,\n","               -1.8588643e-04,  2.2427371e-04,  6.0353847e-04],\n","              [-5.7366857e-04, -1.7669093e-04, -6.3450287e-05, ...,\n","                3.2664344e-04,  1.4837690e-04,  2.1806629e-04],\n","              [ 2.4955563e-04, -4.0498388e-04, -2.0864554e-04, ...,\n","               -6.7594025e-04,  1.5429535e-04,  1.0032643e-04],\n","              ...,\n","              [-1.9578889e-04, -6.7586283e-04,  6.7986274e-04, ...,\n","               -1.0873551e-03, -5.8892084e-04,  2.3307976e-04],\n","              [ 4.5623721e-04, -4.8884918e-04, -5.2124931e-04, ...,\n","               -6.0640352e-05,  1.3974101e-04, -3.7324586e-04],\n","              [-2.6225662e-04,  6.9012016e-04, -1.1878642e-04, ...,\n","               -3.5870206e-04, -2.5932130e-04, -1.1665131e-04]], dtype=float32),\n","       array([-0.00026358, -0.00018778, -0.00029837, ..., -0.00035721,\n","              -0.00020364, -0.00023647], dtype=float32)),\n","      ((), (), ()),\n","      (),\n","      (array([[-1.0623553e-04,  5.6222132e-05, -8.6368447e-05, ...,\n","                4.7722300e-05, -4.0235472e-04, -3.4111028e-05],\n","              [ 4.2976963e-04,  5.9981008e-05, -9.1083610e-05, ...,\n","               -1.8995373e-05,  3.4361114e-05, -6.2762556e-04],\n","              [-4.7806939e-04,  2.7012484e-04,  4.0250691e-04, ...,\n","               -8.5164356e-04, -4.5572463e-04, -7.6741539e-04],\n","              ...,\n","              [-3.6244441e-04, -3.9228128e-04,  1.6872550e-04, ...,\n","                4.4656510e-05,  3.0676849e-05,  2.5184228e-04],\n","              [ 6.6042514e-05, -8.6441520e-05, -7.7052187e-05, ...,\n","               -7.0401533e-05, -1.7572983e-04,  8.4946043e-04],\n","              [ 3.5573514e-05,  7.2914839e-04,  1.8940045e-04, ...,\n","               -1.3019874e-03, -4.6824709e-05,  5.7730917e-04]], dtype=float32),\n","       array([-1.2910289e-04, -4.9095506e-05,  3.7151235e-04, ...,\n","               2.3424672e-04, -1.7630338e-04, -4.3283735e-04], dtype=float32)),\n","      ()))),\n","   ()),\n","  (((),\n","    ((),\n","     ((array([ 9.909098 , 11.428105 ,  9.799679 , ...,  7.9962783, 10.893047 ,\n","              10.1048   ], dtype=float32),\n","       array([ 0.00065552, -0.00037794,  0.00030654, ...,  0.00059264,\n","               0.00038433, -0.00017869], dtype=float32)),\n","      ((),\n","       ((),\n","        (((),\n","          (((array([[ 1.80963136e-04, -1.72411755e-03,  2.53614620e-04, ...,\n","                      6.66238135e-04, -5.64404880e-04,  4.34598594e-04],\n","                    [ 3.29284725e-04,  1.67489750e-04, -6.97102398e-04, ...,\n","                      7.33508379e-04, -5.49852033e-04, -8.90243682e-05],\n","                    [-6.37591642e-04,  1.59089861e-03,  3.30858602e-04, ...,\n","                      2.54959130e-04, -6.68039313e-04,  8.96985730e-05],\n","                    ...,\n","                    [ 3.53550451e-04, -1.45589647e-05, -5.98424580e-04, ...,\n","                     -1.00866615e-04,  7.16681418e-04,  3.79509176e-04],\n","                    [-4.82743693e-04, -3.72535258e-04, -7.85693373e-06, ...,\n","                     -9.85914492e-04,  1.45347629e-04,  1.47652943e-04],\n","                    [-5.29564277e-04, -8.38685082e-05,  4.84178570e-04, ...,\n","                      7.00469071e-04, -7.37989787e-04, -6.10083080e-05]], dtype=float32),\n","             array([-2.5801503e-04, -2.7463012e-04,  3.1022017e-04, ...,\n","                    -3.5322050e-04,  5.6987966e-04, -4.2092575e-05], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[ 6.1109278e-04, -1.1260682e-03,  4.7351356e-04, ...,\n","                     -8.8380295e-04,  1.2926895e-03, -1.9283389e-04],\n","                    [-1.6222277e-03, -1.8947807e-04,  1.4476199e-03, ...,\n","                      1.2111663e-03,  4.1745038e-04, -1.4482756e-04],\n","                    [-1.0165056e-03,  2.2282647e-03,  4.7706431e-04, ...,\n","                     -2.8472819e-04,  2.0597728e-04, -4.0326078e-04],\n","                    ...,\n","                    [ 5.0623005e-04,  3.5074266e-04, -4.2140900e-04, ...,\n","                     -1.2938341e-04,  5.2355154e-04,  4.6946623e-04],\n","                    [ 8.0375391e-04, -1.7217009e-03, -1.9867492e-03, ...,\n","                     -1.5799197e-03, -5.1825169e-05, -4.3434222e-04],\n","                    [ 1.0776016e-03, -2.0183813e-04,  2.5813360e-04, ...,\n","                     -3.9201157e-04,  1.6555196e-04, -5.4713013e-04]], dtype=float32),\n","             array([ 4.0247355e-04, -6.3286792e-04, -5.5743389e-05, ...,\n","                     3.6390350e-04,  5.5498205e-04,  4.0642580e-04], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[-9.2115542e-06, -3.7608959e-04, -2.1323582e-04, ...,\n","                     -2.4493167e-04,  3.2800762e-04, -1.9525072e-04],\n","                    [-1.2070422e-04,  2.3351595e-04,  2.5672620e-04, ...,\n","                      2.9504692e-04, -5.3461839e-04, -9.8853270e-05],\n","                    [ 2.1171370e-04, -7.1005328e-05, -1.2319130e-04, ...,\n","                     -4.0038073e-04, -4.6292326e-04, -7.8322802e-05],\n","                    ...,\n","                    [-5.4718152e-04, -2.8939662e-04, -2.4404180e-04, ...,\n","                      3.6225581e-04,  7.9589646e-04,  8.5793719e-05],\n","                    [-2.0745186e-04,  4.9271566e-05,  1.4059735e-04, ...,\n","                     -5.0729769e-04, -3.3909167e-04, -9.0322224e-05],\n","                    [ 8.1274433e-05, -1.8218235e-04, -2.0270234e-05, ...,\n","                      1.7343904e-04, -7.1452487e-05,  3.1430717e-04]], dtype=float32),\n","             array([-6.4741071e-05, -3.0238522e-04,  4.6613233e-05, ...,\n","                    -5.0222981e-05,  4.9745267e-05, -1.1149130e-04], dtype=float32)),\n","            ((), (), ())))),\n","         (),\n","         ((), (), ()),\n","         (array([[-2.4250365e-04, -3.1713763e-04,  1.7700398e-04, ...,\n","                  -2.6153272e-04,  3.0795706e-04, -1.0494300e-04],\n","                 [ 3.8190334e-04, -1.8679537e-04, -3.8757827e-04, ...,\n","                   5.8713648e-04,  2.9448909e-04,  6.9269777e-04],\n","                 [ 1.6848408e-04, -3.7480309e-05, -2.2980885e-05, ...,\n","                   8.2943598e-06,  8.7178021e-05,  3.0045343e-05],\n","                 ...,\n","                 [-2.6248375e-04, -7.8549969e-04, -3.7278593e-04, ...,\n","                   2.4296733e-04,  1.3783952e-04, -3.4079215e-04],\n","                 [ 3.7385552e-04, -2.5290638e-04, -2.3546227e-04, ...,\n","                  -2.0059713e-04,  4.4951760e-04, -1.1169907e-04],\n","                 [-2.9794476e-04, -8.4281965e-05, -5.0073536e-04, ...,\n","                  -5.3642690e-04,  1.6828071e-04, -9.3656607e-05]], dtype=float32),\n","          array([-1.7873319e-04, -1.5377047e-05,  2.3824688e-04, ...,\n","                  5.0577444e-05,  1.0308183e-05,  4.8045837e-05], dtype=float32))),\n","        ()),\n","       ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([27.666683, 24.298004, 28.497082, ..., 25.02334 , 27.010147,\n","              21.95867 ], dtype=float32),\n","       array([-2.2025406e-04, -3.2912189e-04, -1.8263828e-04, ...,\n","               5.2202365e-04, -3.7271847e-04,  7.0229216e-05], dtype=float32)),\n","      (),\n","      ((), (((array([[ 3.17935395e-04, -7.82662712e-04, -6.45369000e-04, ...,\n","                  -7.57013098e-04,  7.02629331e-04, -1.08367682e-03],\n","                 [ 8.39802204e-04, -1.24313624e-03, -8.43868474e-05, ...,\n","                  -1.04953861e-03,  2.76605424e-04,  6.51801005e-04],\n","                 [ 1.58111530e-03,  6.71174203e-04, -5.59988723e-04, ...,\n","                   6.32068259e-04, -1.05244631e-03, -5.58039232e-04],\n","                 ...,\n","                 [ 1.18595606e-04, -3.39220307e-04,  8.70200165e-04, ...,\n","                   1.30216673e-03,  6.62511156e-04,  9.16509714e-04],\n","                 [-3.39543272e-04,  1.94550911e-03,  1.15994051e-04, ...,\n","                   8.55145452e-04, -4.06545383e-04,  1.47446801e-04],\n","                 [ 3.89938185e-04,  4.08718275e-04,  5.01883515e-05, ...,\n","                  -5.57583466e-04,  1.65818055e-04,  2.57118343e-04]], dtype=float32),\n","          array([-5.0125884e-05, -7.6582539e-05,  2.9356687e-04, ...,\n","                  5.0109014e-04, -3.9477262e-04,  3.9145787e-04], dtype=float32)),\n","         ((array([[ 1.4259289e-03, -1.1663976e-03, -6.0275645e-04, ...,\n","                    6.5766938e-04, -1.6165720e-03, -4.9851200e-04],\n","                  [ 5.3573068e-04, -3.4731551e-04,  7.3190808e-04, ...,\n","                    7.8739738e-04, -2.8049544e-04, -3.4237048e-04],\n","                  [ 4.6062321e-04,  6.6937460e-04, -6.7506835e-04, ...,\n","                    6.6987018e-04, -1.6523544e-04, -5.8827805e-04],\n","                  ...,\n","                  [-4.2613095e-04, -1.7639896e-04, -8.6231064e-04, ...,\n","                   -1.5014902e-04, -5.6814519e-04, -1.1350592e-03],\n","                  [-1.0009821e-03, -4.0471173e-04,  8.6628112e-05, ...,\n","                    2.0653555e-04, -2.1214850e-04, -1.3493400e-03],\n","                  [-5.8538248e-05,  1.3810501e-03,  8.2403974e-04, ...,\n","                    1.2826193e-03, -1.6831111e-03,  1.3137193e-03]], dtype=float32),\n","           array([-6.9449678e-05, -6.2366034e-06, -1.4306141e-04, ...,\n","                   4.2092515e-04, -4.9613230e-04,  5.5778067e-04], dtype=float32)),),\n","         ((array([[ 6.2881835e-04,  5.2332709e-04,  1.8533703e-04, ...,\n","                   -1.3407919e-05,  5.5843056e-04,  7.6834497e-04],\n","                  [ 5.1182113e-04,  5.7660608e-04,  3.0054021e-04, ...,\n","                    7.3896867e-05, -1.7557539e-04, -8.6865723e-05],\n","                  [ 4.7844287e-04,  2.5775464e-04, -6.2255480e-05, ...,\n","                    4.2353809e-04, -1.4554513e-04,  2.9171564e-04],\n","                  ...,\n","                  [ 3.6863852e-04,  5.8157521e-04, -9.0340523e-05, ...,\n","                    1.9225407e-04,  5.2006938e-04, -4.1247270e-04],\n","                  [ 2.6453141e-05,  6.8822625e-04,  1.6925461e-04, ...,\n","                    1.4226996e-04,  2.4623799e-04,  4.9875071e-04],\n","                  [ 8.8601133e-05,  8.9200260e-04, -5.1677175e-04, ...,\n","                   -3.9046776e-05,  2.4477333e-06,  3.6193486e-04]], dtype=float32),\n","           array([-2.0951711e-04, -1.5966715e-04, -9.0573449e-05, ...,\n","                   1.6809560e-04,  2.4261269e-04,  7.7525285e-05], dtype=float32)),)),\n","        (),\n","        (array([[ 2.4527480e-04,  3.4490437e-04,  3.0943810e-04, ...,\n","                  4.5112587e-04,  2.1346626e-04, -2.6866666e-04],\n","                [-3.7744129e-04,  2.5811463e-04, -2.1393242e-04, ...,\n","                 -3.9314702e-05, -3.6374084e-05,  7.6267915e-04],\n","                [-3.8383889e-04,  1.1842908e-04, -2.2704345e-04, ...,\n","                 -3.3887970e-04, -5.2842341e-04,  2.0263893e-05],\n","                ...,\n","                [ 2.0519944e-04, -1.3071859e-04,  4.3726928e-04, ...,\n","                  3.6562019e-04, -1.7234050e-04, -4.5263361e-05],\n","                [ 2.1188581e-04,  2.9319624e-04, -3.4948954e-04, ...,\n","                 -4.3392839e-04, -5.8998558e-04, -8.0334203e-04],\n","                [ 4.0151988e-04,  2.5214755e-04, -1.3229705e-04, ...,\n","                  2.9866080e-04,  1.9303303e-04, -3.0595291e-04]], dtype=float32),\n","         array([-1.7921485e-04,  5.4273729e-05,  2.4546593e-04, ...,\n","                -6.6922170e-05, -5.8287795e-05,  7.2198862e-05], dtype=float32))), ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([33.013134, 26.700388, 32.309517, ..., 31.560844, 27.547998,\n","              33.83545 ], dtype=float32),\n","       array([-0.00233126,  0.0039384 , -0.00223921, ..., -0.0011619 ,\n","              -0.0005185 ,  0.00673685], dtype=float32)),\n","      (array([[ 3.2771449e-04, -4.0412026e-05, -2.4454324e-05, ...,\n","               -9.6628314e-04,  6.3420594e-05, -4.8846308e-05],\n","              [ 1.1286322e-05,  3.0727050e-04,  1.2533183e-04, ...,\n","                1.0583549e-03,  2.9930804e-04,  6.9362133e-05],\n","              [-1.3207090e-04, -4.3077770e-04, -4.8681884e-04, ...,\n","               -3.7943060e-04,  2.3240880e-04,  1.2930224e-04],\n","              ...,\n","              [ 4.9480482e-04,  2.5664151e-04, -2.6073688e-04, ...,\n","               -2.8287986e-04,  3.2067590e-04, -1.4706752e-04],\n","              [-3.3403773e-04, -1.3719923e-04, -1.8442736e-04, ...,\n","                3.4763457e-04,  4.9896349e-05, -1.0417709e-04],\n","              [ 2.6831287e-04, -5.2795961e-04,  3.9545176e-04, ...,\n","               -1.0107345e-04, -5.6582049e-04,  3.1664787e-04]], dtype=float32),\n","       array([-0.00020462, -0.00016675, -0.00021529, ..., -0.00022891,\n","              -0.00023212, -0.0002708 ], dtype=float32)),\n","      ((), (), ()),\n","      (),\n","      (array([[-5.0125946e-04,  6.4438704e-04, -3.2812485e-04, ...,\n","                9.6950153e-06,  6.4625102e-04,  5.0906261e-04],\n","              [-6.7125051e-04,  5.1556062e-04, -5.1852444e-04, ...,\n","               -2.7532189e-04, -1.6568312e-04,  5.6345970e-04],\n","              [ 1.5048563e-04,  2.3399205e-04,  7.8852888e-04, ...,\n","               -2.2977730e-04,  6.9212075e-04, -7.3649024e-04],\n","              ...,\n","              [-9.9023734e-04,  2.0657122e-04,  3.3111349e-04, ...,\n","               -3.8692303e-04,  9.8248047e-04,  5.5454910e-04],\n","              [-5.5613299e-04, -1.7161547e-04,  8.1493304e-04, ...,\n","               -2.0285173e-04, -8.6443831e-04, -9.7070896e-04],\n","              [ 1.9301666e-04, -1.2331316e-04, -2.7228054e-04, ...,\n","                4.0160998e-04,  4.6902490e-04,  6.1422714e-04]], dtype=float32),\n","       array([-1.6426903e-04, -5.2797935e-05,  3.3303167e-04, ...,\n","               1.7356148e-04, -1.7477297e-04, -1.9834703e-04], dtype=float32)),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([16.3547  , 22.156084, 23.084677, ..., 19.758577, 23.530777,\n","              22.051527], dtype=float32),\n","       array([-7.841374e-06,  6.333542e-05,  5.021927e-04, ...,  7.787898e-05,\n","               2.911078e-04, -2.337846e-04], dtype=float32)),\n","      ((),\n","       ((),\n","        (((), (((array([[ 6.1483246e-05,  7.4685382e-04,  6.8871590e-04, ...,\n","                      3.1261309e-04, -2.1767468e-04, -5.2827294e-04],\n","                    [-5.0699422e-05,  1.1016885e-03,  1.1913941e-03, ...,\n","                     -1.2116972e-04,  8.4310095e-04, -2.1788730e-04],\n","                    [-1.1305427e-04, -2.5385417e-04, -6.0314784e-04, ...,\n","                     -2.1491440e-03, -3.8612686e-04,  5.7008292e-04],\n","                    ...,\n","                    [ 5.7249743e-04,  4.6029006e-04,  2.5812466e-05, ...,\n","                     -1.5434396e-04, -8.8169554e-04, -8.6321332e-04],\n","                    [-7.3259551e-05, -1.2874920e-03,  8.6988628e-05, ...,\n","                     -3.6276400e-05,  9.8776061e-04,  1.2623394e-03],\n","                    [ 5.0609885e-04,  2.2816541e-05, -1.0334901e-03, ...,\n","                     -5.6002440e-04,  3.6289261e-05, -6.3421490e-04]], dtype=float32),\n","             array([-4.7778216e-04,  6.8889931e-05,  2.0761880e-04, ...,\n","                     4.5873996e-05,  2.7833674e-05, -1.4248873e-04], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[-1.3746934e-03,  1.6175075e-03, -6.1864924e-04, ...,\n","                     -3.3792556e-04, -5.4738496e-04, -2.0109327e-04],\n","                    [-2.9143042e-04,  6.1741885e-05,  1.1154297e-03, ...,\n","                     -9.6768449e-04,  9.4610688e-05,  4.3283537e-04],\n","                    [ 7.8726065e-05, -5.1035645e-04,  1.0544629e-03, ...,\n","                     -1.2011930e-03, -1.0433214e-03,  1.3448813e-03],\n","                    ...,\n","                    [ 3.3177118e-04,  8.6553872e-04,  2.1796160e-04, ...,\n","                     -8.3672418e-04, -2.9772962e-04, -1.1818978e-03],\n","                    [ 3.8248944e-04, -1.4083424e-03,  1.4428345e-04, ...,\n","                     -5.0316652e-04,  8.5582089e-04,  1.9056363e-03],\n","                    [ 6.9525040e-04, -2.9183499e-05, -5.1701302e-04, ...,\n","                     -4.6823814e-04, -3.4060757e-04,  1.1953153e-04]], dtype=float32),\n","             array([ 3.1094509e-04,  5.3171004e-04, -1.8410418e-04, ...,\n","                     2.3028645e-04, -4.2940930e-05, -4.1093098e-04], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[ 3.64723710e-05,  3.06765105e-05, -2.88179843e-04, ...,\n","                      3.06271191e-04, -2.76296400e-04,  1.84498931e-04],\n","                    [-2.48657685e-04, -4.11833113e-04, -1.58469062e-04, ...,\n","                      4.47541359e-04, -5.60549146e-04, -5.33064944e-04],\n","                    [-9.34662821e-05,  1.93048822e-04,  7.41521828e-04, ...,\n","                      1.99062444e-04, -6.00227970e-04,  1.69384410e-04],\n","                    ...,\n","                    [ 1.13372924e-04, -1.31809822e-04, -2.47532793e-04, ...,\n","                      2.72990350e-04,  9.91128472e-05, -1.31201741e-04],\n","                    [ 7.05009268e-04,  3.53255455e-04,  4.17664793e-04, ...,\n","                     -1.31453227e-04,  2.23351781e-05, -3.50277784e-04],\n","                    [-2.01161078e-04, -1.64198689e-04, -4.02835140e-04, ...,\n","                     -1.45295024e-04,  7.15239294e-05,  2.35098953e-04]], dtype=float32),\n","             array([-2.0216451e-04,  2.1548517e-04,  7.1255432e-05, ...,\n","                     1.9430165e-04,  2.6130539e-04, -7.3771458e-05], dtype=float32)),\n","            ((), (), ())))),\n","         (),\n","         ((), (), ()),\n","         (array([[-8.77280254e-05,  6.43201696e-04,  1.15211384e-04, ...,\n","                  -4.93474479e-04, -2.04726981e-04,  5.79610176e-04],\n","                 [ 7.92369519e-06, -3.79015226e-04,  4.72584681e-04, ...,\n","                   7.74941349e-04, -3.46693851e-04,  4.36632283e-04],\n","                 [-3.34304845e-04,  1.43579455e-04, -8.06040334e-05, ...,\n","                   8.82788299e-05, -1.43980084e-04,  5.17989902e-05],\n","                 ...,\n","                 [ 1.84513512e-04, -4.46527411e-04,  4.40933538e-04, ...,\n","                  -7.02700781e-05, -9.45490436e-04,  6.48573696e-05],\n","                 [-6.88771433e-06, -1.50292748e-04,  7.63986376e-04, ...,\n","                   2.57567499e-05,  7.36119691e-05,  1.02562772e-04],\n","                 [-7.93892177e-05, -4.65895282e-05, -7.38648116e-04, ...,\n","                   3.37358251e-05, -2.74577207e-04, -1.28141721e-04]], dtype=float32),\n","          array([-2.2618379e-04,  9.4455805e-05,  1.7288784e-04, ...,\n","                  4.4897544e-05,  6.3861349e-05,  1.3886786e-06], dtype=float32))),\n","        ()),\n","       ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([23.88225 , 24.184444, 24.261084, ..., 19.43912 , 23.9153  ,\n","              21.453533], dtype=float32),\n","       array([-7.1385072e-04, -5.4409995e-04,  2.4066308e-04, ...,\n","               6.9122034e-04, -2.5002286e-04,  1.0099970e-05], dtype=float32)),\n","      (),\n","      ((), (((array([[ 3.6466459e-04,  2.1552869e-04,  9.6790624e-05, ...,\n","                   8.0245500e-04, -3.5085523e-04, -1.9914127e-04],\n","                 [-1.3182993e-03,  2.5995169e-04,  7.5160220e-05, ...,\n","                  -3.9431948e-04, -4.6758665e-04, -8.5796311e-04],\n","                 [-4.7536272e-05, -4.8585830e-04,  7.3798881e-05, ...,\n","                   8.8379678e-04,  9.1281586e-04,  1.1242417e-03],\n","                 ...,\n","                 [-3.9362403e-05, -1.0642599e-03,  6.7456887e-04, ...,\n","                  -3.6807364e-04, -2.7723247e-04,  7.8843971e-04],\n","                 [ 1.6825569e-04,  3.9978636e-06, -1.1445450e-03, ...,\n","                   5.2205485e-04, -4.7387229e-04,  4.4650357e-04],\n","                 [-1.0540227e-03,  7.4064126e-04,  8.8554103e-04, ...,\n","                  -6.7325815e-04, -3.3982516e-05, -4.3386099e-04]], dtype=float32),\n","          array([ 0.00011217,  0.00012424, -0.00022637, ..., -0.00025177,\n","                  0.00022989, -0.00021651], dtype=float32)),\n","         ((array([[-3.6026686e-04,  2.3753387e-04, -1.6272849e-04, ...,\n","                    1.7211484e-03,  3.2584078e-04, -1.4673040e-03],\n","                  [-1.0720923e-03, -4.8753052e-04,  4.9335722e-05, ...,\n","                   -1.3573533e-04, -1.0209627e-03,  4.9405376e-04],\n","                  [ 9.0015790e-04, -5.5950414e-04,  8.2321541e-04, ...,\n","                   -5.8994209e-04, -2.1278909e-04, -5.9460063e-04],\n","                  ...,\n","                  [-2.0530834e-04, -1.3719196e-04,  1.6525913e-03, ...,\n","                   -1.0569103e-03,  1.9768473e-04,  1.0838987e-03],\n","                  [ 7.8881090e-04, -8.2777947e-04, -1.0765583e-03, ...,\n","                   -5.4017402e-04,  5.9777912e-04, -5.7033205e-04],\n","                  [ 5.0317508e-04,  2.5128157e-04,  5.0119753e-04, ...,\n","                   -8.6353405e-04,  9.7763550e-04, -2.1591253e-04]], dtype=float32),\n","           array([-7.2369352e-05, -1.7596563e-04, -4.8424781e-04, ...,\n","                   1.0572466e-04,  1.2420473e-04, -8.3882059e-04], dtype=float32)),),\n","         ((array([[ 4.1991376e-04,  2.2568824e-04,  2.9331533e-04, ...,\n","                    5.9546879e-04, -1.9565303e-05, -5.8844319e-04],\n","                  [ 4.3019193e-04, -9.5008785e-04,  1.5475822e-04, ...,\n","                   -4.0054115e-04, -5.6435302e-04,  8.9933042e-04],\n","                  [-2.8103942e-04, -1.5729204e-04, -1.8776223e-04, ...,\n","                    2.8744049e-04,  1.5519127e-04, -7.5541548e-06],\n","                  ...,\n","                  [-3.3508264e-04, -5.3719443e-04,  4.1837839e-04, ...,\n","                    1.6026276e-04, -1.5685641e-03, -2.2902989e-04],\n","                  [ 8.9147070e-04, -2.6091552e-04,  2.1424978e-04, ...,\n","                    5.8429409e-04, -4.9642072e-04, -9.5681881e-04],\n","                  [-6.3942868e-04, -4.0233121e-04,  6.0142887e-05, ...,\n","                   -6.4284698e-04, -5.8053259e-04, -7.0160779e-04]], dtype=float32),\n","           array([ 6.4041204e-05,  1.1700615e-05, -3.9535826e-05, ...,\n","                  -4.5969060e-05,  5.2684807e-04, -1.2541121e-04], dtype=float32)),)),\n","        (),\n","        (array([[-1.8251000e-04,  4.7497710e-04,  4.1154795e-04, ...,\n","                  3.1851072e-04, -1.0820047e-04, -5.6373392e-04],\n","                [-1.8394567e-04,  7.2167628e-04, -1.3643969e-03, ...,\n","                 -1.4015054e-04, -7.7690353e-04,  6.8640875e-05],\n","                [ 2.5754597e-04,  2.7684707e-04, -3.2690514e-04, ...,\n","                 -2.5374908e-04, -2.3440904e-05,  2.3832501e-04],\n","                ...,\n","                [-5.0068880e-04, -4.7192058e-05, -6.1835756e-04, ...,\n","                  3.0329567e-04, -2.0094462e-04,  1.4722493e-04],\n","                [-9.7752418e-05,  3.8573201e-04,  4.3632067e-04, ...,\n","                 -5.7262889e-05, -5.7426194e-04,  7.7493663e-04],\n","                [ 4.6315580e-04,  6.8317907e-04, -3.2689775e-04, ...,\n","                 -1.9134724e-04, -1.8184795e-04,  4.8428893e-04]], dtype=float32),\n","         array([-5.6365488e-06,  2.4791760e-05,  2.1130762e-04, ...,\n","                -6.0887119e-06, -8.4632389e-05,  1.3121450e-04], dtype=float32))), ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([39.219814, 32.74529 , 38.38454 , ..., 36.090923, 30.186974,\n","              36.929134], dtype=float32),\n","       array([-0.00412493,  0.00391778, -0.00232285, ..., -0.00211273,\n","              -0.00197973,  0.00526257], dtype=float32)),\n","      (array([[ 3.2264111e-04,  3.9771137e-05, -4.4960107e-04, ...,\n","                2.1897566e-05,  3.2064816e-04,  3.9232746e-04],\n","              [-6.7474361e-04,  2.9992993e-04, -1.6200924e-05, ...,\n","                4.9566779e-06, -2.1608520e-04,  5.5354998e-05],\n","              [ 3.8022752e-04,  2.6469948e-04, -1.5772102e-04, ...,\n","               -1.9269937e-04, -3.1495589e-04, -1.5626263e-05],\n","              ...,\n","              [-1.9471291e-04, -3.6615183e-04, -7.8032986e-05, ...,\n","                8.2358712e-04,  1.6707313e-04, -3.8224537e-04],\n","              [-2.8016116e-04, -3.5487625e-04, -2.7940619e-05, ...,\n","                5.5918592e-04, -6.7758039e-05, -4.9956702e-04],\n","              [-4.5153179e-04, -1.8954619e-04,  4.0361623e-04, ...,\n","                1.0671399e-03, -1.2776767e-03, -3.5355816e-04]], dtype=float32),\n","       array([-0.0001813 , -0.00018283, -0.00023269, ..., -0.00021104,\n","              -0.00037288, -0.00023105], dtype=float32)),\n","      ((), (), ()),\n","      (),\n","      (array([[-6.07041526e-04, -5.89260133e-04,  1.91954503e-04, ...,\n","               -6.51813811e-04, -3.63248328e-05, -6.18508144e-04],\n","              [ 1.21632744e-04,  3.12897784e-04, -2.95754726e-05, ...,\n","                3.63708386e-04, -5.60515269e-04, -5.78314124e-04],\n","              [ 2.87323288e-04, -2.56691827e-04, -3.61674785e-04, ...,\n","                3.02339147e-04, -7.79822803e-05,  2.60879606e-04],\n","              ...,\n","              [-2.72708712e-04, -3.09692346e-04, -6.54434261e-04, ...,\n","               -6.56898308e-04,  7.99924892e-04, -1.05635525e-04],\n","              [ 8.76000733e-04,  7.94008607e-04,  5.45370101e-04, ...,\n","               -2.97376100e-04,  7.29779771e-04, -8.74122907e-06],\n","              [-1.87172729e-04,  9.88529282e-05, -1.42348377e-04, ...,\n","                7.11906643e-04, -1.71261374e-04, -8.82027234e-05]], dtype=float32),\n","       array([ 0.00012392, -0.00028158,  0.00034114, ...,  0.00014256,\n","              -0.0001409 , -0.00032786], dtype=float32)),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([13.221839, 15.507087, 15.17562 , ..., 15.518029, 18.71215 ,\n","              15.768428], dtype=float32),\n","       array([ 5.6227646e-04,  1.3988595e-04,  5.1316043e-04, ...,\n","               1.6706108e-04,  6.8293710e-04, -6.1461811e-05], dtype=float32)),\n","      ((),\n","       ((),\n","        (((),\n","          (((array([[ 2.79998843e-04, -1.07051141e-03, -3.15049372e-04, ...,\n","                      2.12975749e-04,  1.02279149e-03, -4.38018498e-04],\n","                    [-3.27870599e-04, -1.92390013e-04, -3.41603416e-04, ...,\n","                      1.42549002e-03, -1.46539547e-04,  7.56135443e-04],\n","                    [-1.14563358e-04,  1.08892695e-04,  1.11117819e-03, ...,\n","                      6.52332848e-04,  1.33359258e-03, -2.06675089e-04],\n","                    ...,\n","                    [-4.17379051e-04,  8.06581578e-04, -1.45165657e-04, ...,\n","                     -1.30351284e-03,  6.43594423e-04,  1.63201457e-05],\n","                    [-3.93084832e-04, -1.38887845e-04,  4.45881829e-04, ...,\n","                     -1.39148353e-04, -7.23995618e-04, -1.68725135e-04],\n","                    [ 9.90677741e-04, -2.91209697e-04, -2.76336446e-04, ...,\n","                     -2.38632807e-03,  3.01859225e-04, -7.11652567e-04]], dtype=float32),\n","             array([ 1.6056283e-04, -1.7568443e-04,  1.4395774e-06, ...,\n","                    -5.9091288e-04,  3.8078395e-04,  1.2718572e-04], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[ 5.3045049e-04, -8.8599819e-04, -6.5095533e-06, ...,\n","                     -2.0052481e-04,  7.4881385e-04, -1.1153152e-04],\n","                    [ 7.0103951e-04, -1.4661106e-04, -5.4170919e-04, ...,\n","                     -1.8124770e-04,  2.6543025e-04,  4.4295008e-04],\n","                    [-3.7302449e-04, -4.4398065e-04,  8.7001844e-04, ...,\n","                      2.9667860e-04,  1.4250837e-03,  7.4736861e-04],\n","                    ...,\n","                    [-4.3023285e-04,  9.5379550e-04,  6.2287523e-04, ...,\n","                     -5.7794346e-04,  1.9658432e-07,  2.5717166e-04],\n","                    [-1.4114323e-04,  2.0276009e-04, -4.5244582e-04, ...,\n","                     -4.9241347e-04, -4.8593158e-04,  3.1854163e-04],\n","                    [ 4.8871945e-05, -5.8529066e-04,  7.2798639e-04, ...,\n","                      3.7680031e-05,  9.0196863e-04,  2.0063353e-04]], dtype=float32),\n","             array([-2.2388790e-04,  2.3948469e-05,  3.8035435e-04, ...,\n","                    -1.0276325e-04, -8.8133544e-05,  3.0947594e-05], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[-8.03980947e-05,  4.02495411e-04, -5.34943014e-04, ...,\n","                      5.49706223e-04, -3.71957431e-04, -7.43918063e-05],\n","                    [-1.58995696e-04, -8.04227602e-04,  5.75686747e-04, ...,\n","                      2.36024367e-04, -1.95321816e-04,  1.46337325e-05],\n","                    [-4.13309346e-04, -7.86876190e-04, -6.89146749e-04, ...,\n","                      8.03973817e-04, -3.64478241e-04,  9.87983425e-04],\n","                    ...,\n","                    [ 6.41172082e-05, -9.15527926e-05, -9.29834714e-05, ...,\n","                      5.86092472e-04, -3.41504317e-04, -2.23352632e-04],\n","                    [ 2.53275211e-05,  2.97678402e-04,  1.57185801e-04, ...,\n","                     -1.73145716e-04, -7.21550838e-04,  5.85008893e-05],\n","                    [-3.87652244e-06,  3.32596741e-04, -4.67848527e-04, ...,\n","                     -3.56841163e-04, -6.60274411e-04,  1.63282602e-04]], dtype=float32),\n","             array([-1.0055124e-04,  1.5575042e-05,  8.0013844e-05, ...,\n","                    -4.4125201e-05, -1.1925846e-04,  1.5062957e-04], dtype=float32)),\n","            ((), (), ())))),\n","         (),\n","         ((), (), ()),\n","         (array([[ 7.6520279e-05,  4.5464774e-05,  5.0756754e-04, ...,\n","                  -1.0289553e-04, -7.4671072e-05, -5.5004243e-04],\n","                 [-3.5025767e-04,  9.1569405e-04,  3.6493069e-04, ...,\n","                   2.4842791e-04, -3.3111635e-04, -2.3453268e-04],\n","                 [ 2.8800988e-04, -9.0527709e-04,  3.2245702e-04, ...,\n","                  -7.8086631e-04, -1.5940527e-04, -1.9475308e-05],\n","                 ...,\n","                 [ 2.9549908e-04, -4.4733332e-04,  5.0390558e-04, ...,\n","                   1.0353600e-04,  8.2045217e-04, -1.4252569e-04],\n","                 [ 3.8873495e-05,  2.1729537e-04, -8.4105544e-05, ...,\n","                   5.6474609e-04, -5.3949811e-04, -1.3547888e-03],\n","                 [-3.8063903e-05, -2.7078469e-04,  4.9355748e-04, ...,\n","                   1.8791163e-04, -3.6855141e-04,  2.0029415e-04]], dtype=float32),\n","          array([-2.4778873e-04, -6.9160822e-05,  2.2130902e-04, ...,\n","                 -5.3707183e-05, -5.9509475e-05, -1.0200243e-06], dtype=float32))),\n","        ()),\n","       ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([39.217216, 34.7438  , 44.571598, ..., 43.651512, 41.88316 ,\n","              38.881588], dtype=float32),\n","       array([ 3.1261938e-04,  7.2930338e-06, -4.0973162e-05, ...,\n","               4.6519624e-05, -1.0626103e-05, -2.2409364e-04], dtype=float32)),\n","      (),\n","      ((), (((array([[ 4.36439259e-05,  1.09830382e-03,  5.02440904e-04, ...,\n","                   1.62896264e-04,  2.97324441e-04,  3.38836631e-04],\n","                 [-4.67220641e-04,  4.11740766e-04, -3.33559758e-04, ...,\n","                   5.41153888e-04, -5.04517346e-04, -7.85686134e-04],\n","                 [-1.51019311e-03,  2.03796269e-04, -7.17730436e-04, ...,\n","                   7.42270495e-04, -1.59003233e-04,  6.28769689e-04],\n","                 ...,\n","                 [ 5.26434684e-04,  1.94701977e-04, -5.47865406e-04, ...,\n","                   3.28295835e-04, -3.31033989e-05,  4.81234543e-04],\n","                 [-4.48694365e-04,  4.16946888e-04,  8.60233675e-04, ...,\n","                  -1.04512656e-04, -1.22383458e-03, -1.19001335e-04],\n","                 [ 1.30523142e-04,  3.18305596e-04, -1.31059915e-03, ...,\n","                  -9.92811867e-04, -6.83447579e-04,  7.06704450e-04]], dtype=float32),\n","          array([-2.8142947e-04, -3.5939025e-04,  1.3097549e-05, ...,\n","                  1.1465809e-05, -1.6056097e-04,  7.6249357e-06], dtype=float32)),\n","         ((array([[ 8.3950511e-04, -7.6697161e-04,  4.2048201e-04, ...,\n","                   -1.2229948e-04, -1.8267806e-04,  1.3855977e-03],\n","                  [ 7.6459502e-05, -2.8264121e-04, -8.1469794e-04, ...,\n","                   -2.0935242e-04,  5.7156227e-04, -4.6814739e-04],\n","                  [ 2.2762713e-04, -9.0557318e-05, -4.6329887e-04, ...,\n","                    9.3119859e-04, -3.8856117e-04, -2.3478651e-04],\n","                  ...,\n","                  [ 1.7790889e-04,  4.1472793e-04, -5.8934971e-04, ...,\n","                    7.1088952e-04,  4.9589120e-04,  4.5094849e-04],\n","                  [-6.5881904e-04, -7.2107784e-04,  1.0266342e-04, ...,\n","                   -6.3308806e-04, -1.0451194e-03, -5.0748826e-04],\n","                  [-3.0430342e-04, -9.8070898e-04,  1.2247541e-04, ...,\n","                   -3.0017130e-05, -2.2694991e-04,  1.3715838e-04]], dtype=float32),\n","           array([ 0.00028965,  0.00053934,  0.00028887, ...,  0.00034955,\n","                  -0.00030759,  0.00033562], dtype=float32)),),\n","         ((array([[ 1.3849991e-04, -3.4267231e-04,  8.1239786e-04, ...,\n","                   -4.2655371e-05,  6.3680556e-05,  9.1553877e-05],\n","                  [-1.4399088e-03, -6.3375721e-04,  6.3457934e-05, ...,\n","                    1.1335099e-03, -7.3814171e-04, -5.4862862e-04],\n","                  [ 1.1888964e-04,  1.7821464e-04,  7.1227580e-05, ...,\n","                    3.0098404e-04,  3.1481846e-04,  1.4112768e-05],\n","                  ...,\n","                  [-6.4172456e-04, -6.4903754e-04,  5.9205812e-04, ...,\n","                   -1.3099857e-04, -4.7752983e-04,  5.9261121e-04],\n","                  [-2.7684550e-04,  1.4077419e-04, -3.6988259e-04, ...,\n","                   -2.2865529e-04,  5.6113879e-04,  6.4086745e-04],\n","                  [ 7.1768113e-04, -1.9003329e-04, -2.2389409e-04, ...,\n","                    7.8633497e-04, -1.9229516e-05,  2.6580694e-04]], dtype=float32),\n","           array([9.2884060e-05, 1.2218814e-04, 2.1081189e-04, ..., 1.0933634e-04,\n","                  4.1616822e-04, 1.9882917e-05], dtype=float32)),)),\n","        (),\n","        (array([[ 2.3218371e-04, -1.2964529e-03, -1.2243666e-03, ...,\n","                  6.7512557e-04, -5.3835272e-05, -7.6703116e-05],\n","                [-4.4085828e-04,  3.7566101e-04, -5.6458090e-04, ...,\n","                  1.6339551e-04, -1.7504644e-04, -2.4057522e-04],\n","                [-3.1023787e-04, -2.1723055e-04,  3.4297878e-04, ...,\n","                 -1.4459965e-03,  3.8409224e-04, -6.7862427e-05],\n","                ...,\n","                [ 5.3400325e-04, -6.7154433e-05,  1.0328582e-04, ...,\n","                  5.6361599e-04, -2.1174867e-04, -3.6889274e-04],\n","                [ 4.4912219e-04,  5.5208195e-05,  2.1142401e-04, ...,\n","                  8.0198712e-05, -2.7998930e-04,  8.7470934e-04],\n","                [-7.3380419e-05, -5.7258859e-04,  6.1435276e-05, ...,\n","                  3.8850572e-04,  4.7181307e-05, -9.3546469e-06]], dtype=float32),\n","         array([-1.8310591e-04, -1.6660933e-05,  1.3933373e-04, ...,\n","                 3.9501861e-05, -1.6572907e-04,  1.5333251e-04], dtype=float32))), ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([35.835804, 31.963232, 36.78046 , ..., 36.00661 , 32.235855,\n","              39.63187 ], dtype=float32),\n","       array([-0.00828533,  0.00292211,  0.00047862, ..., -0.00058286,\n","               0.0027672 ,  0.00624749], dtype=float32)),\n","      (array([[ 2.73928512e-04, -1.81391442e-04,  2.98912171e-04, ...,\n","               -9.03852560e-05,  4.11032262e-04, -2.45751784e-04],\n","              [-8.90337629e-04,  4.04192047e-04,  1.05755054e-04, ...,\n","                1.07050895e-04,  4.94863372e-04,  2.63189559e-05],\n","              [-5.79951047e-05, -9.36054203e-05, -5.97707403e-05, ...,\n","                2.79334898e-04, -2.62540503e-04, -3.40517407e-04],\n","              ...,\n","              [ 1.84469769e-04,  4.78537659e-05,  3.33709380e-04, ...,\n","               -3.10000672e-04, -3.18700113e-05, -8.86789567e-05],\n","              [ 1.51760745e-04,  2.22242888e-04, -7.51460757e-05, ...,\n","               -1.49422893e-04, -1.06039224e-03, -2.85205926e-04],\n","              [-7.41155643e-04, -2.12298648e-04, -3.57353856e-04, ...,\n","               -6.47187931e-04, -1.30727165e-03, -3.94356641e-04]], dtype=float32),\n","       array([-0.00016683, -0.00023429, -0.00024278, ..., -0.00024064,\n","              -0.0001399 , -0.00022493], dtype=float32)),\n","      ((), (), ()),\n","      (),\n","      (array([[-1.6312215e-04, -2.0920332e-04,  7.4090471e-04, ...,\n","               -2.5119679e-04,  4.9558996e-05,  1.2424486e-04],\n","              [-6.9675263e-04, -4.3560681e-04, -4.3357117e-04, ...,\n","                7.0692616e-04,  3.4472311e-04, -5.7808025e-04],\n","              [ 2.1429207e-04,  8.1823225e-04, -5.9074833e-04, ...,\n","               -6.2863919e-04,  2.3477946e-06,  6.5094524e-04],\n","              ...,\n","              [ 4.4611408e-04, -5.9706206e-04, -2.4587713e-04, ...,\n","               -4.5666515e-04,  6.2056520e-04, -3.8843855e-04],\n","              [ 4.4920295e-04, -2.5361817e-04, -1.1072685e-03, ...,\n","               -4.7550595e-04,  6.1222212e-04,  9.7393378e-04],\n","              [-8.9201622e-04, -9.3774303e-05, -5.2861014e-04, ...,\n","                1.1068071e-03,  6.7799638e-06, -9.0449525e-04]], dtype=float32),\n","       array([ 6.3957028e-05, -2.1590966e-04,  4.2667380e-04, ...,\n","               2.0514705e-04, -2.7037307e-04, -2.9011263e-04], dtype=float32)),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([25.060698, 21.236645, 25.42332 , ..., 20.135517, 24.722687,\n","              21.971792], dtype=float32),\n","       array([ 4.4960529e-04, -2.3053300e-04,  1.2722838e-04, ...,\n","              -2.6992046e-05,  5.1448640e-04, -8.2327992e-05], dtype=float32)),\n","      ((),\n","       ((),\n","        (((), (((array([[-1.3008052e-03,  7.2619919e-04,  1.5420188e-03, ...,\n","                     -7.9458236e-04,  2.9186197e-04,  3.3146501e-04],\n","                    [-6.2621466e-04, -1.5856235e-04,  5.0983869e-04, ...,\n","                      9.7009994e-05,  3.7151625e-04, -4.7364802e-04],\n","                    [-7.4915081e-04,  8.1569434e-04, -5.4603600e-04, ...,\n","                     -4.9624537e-05, -7.7709404e-04,  4.1739421e-04],\n","                    ...,\n","                    [ 3.5576537e-04, -4.7604757e-04,  3.2706582e-04, ...,\n","                     -2.3238243e-04,  1.3146833e-04,  5.7213550e-04],\n","                    [-1.0986630e-03,  5.5262668e-04,  4.8406475e-04, ...,\n","                      5.1852874e-04, -9.6271484e-04,  1.8463248e-03],\n","                    [ 7.8091619e-04,  1.0044714e-03, -3.0001128e-04, ...,\n","                     -1.6081399e-04, -3.0322422e-04,  1.7065267e-03]], dtype=float32),\n","             array([ 1.04738225e-04, -3.17168218e-04, -1.04565457e-04, ...,\n","                     5.67244715e-04, -1.42482968e-04, -5.88209332e-05], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[-7.0118584e-04,  1.1962772e-03,  4.0193077e-04, ...,\n","                      1.0573569e-03, -1.6764840e-05, -1.2803655e-04],\n","                    [ 1.2200068e-03, -1.3008613e-04, -3.5168731e-04, ...,\n","                      4.2040786e-04,  7.5236516e-04, -1.2126699e-03],\n","                    [-1.2559608e-03, -5.1350531e-04, -6.1426079e-04, ...,\n","                      1.7461181e-04,  6.3729478e-04,  1.0970331e-03],\n","                    ...,\n","                    [-1.8706164e-04, -1.8191808e-03, -2.1022938e-04, ...,\n","                      1.9218631e-04, -1.2760019e-03, -5.1101058e-04],\n","                    [-1.0697423e-04,  6.7780592e-04, -2.0025492e-04, ...,\n","                     -5.1099970e-04,  1.0948203e-03, -4.7229836e-04],\n","                    [ 1.1569131e-03, -2.8635457e-04,  5.6345121e-04, ...,\n","                     -6.0397893e-04, -1.2120736e-03, -1.1703172e-03]], dtype=float32),\n","             array([-4.6585780e-04,  3.5410884e-04, -3.9701344e-04, ...,\n","                     4.0106362e-04, -1.2359886e-04, -3.1823256e-05], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[ 1.2521573e-03, -6.3709868e-04, -1.6721209e-04, ...,\n","                      3.8772824e-04,  4.6525506e-04, -2.6141835e-04],\n","                    [-1.0654140e-03,  8.4652693e-04,  4.3124074e-04, ...,\n","                     -8.2785424e-05, -1.2311200e-04,  3.0129473e-04],\n","                    [ 7.7858487e-05,  1.3543485e-03, -6.2503346e-04, ...,\n","                      2.4701576e-04,  2.1351337e-04, -7.1783672e-04],\n","                    ...,\n","                    [ 3.0464085e-04, -3.8800726e-04,  5.5075396e-04, ...,\n","                      2.6365925e-04, -3.6926317e-04,  3.4671914e-04],\n","                    [-3.0349864e-04, -2.8712754e-04,  4.5322580e-04, ...,\n","                      6.3238316e-04,  3.6455743e-04,  6.4699189e-04],\n","                    [ 7.1890117e-04,  5.1156990e-04, -4.8988429e-04, ...,\n","                     -1.1337268e-03, -1.6186787e-04,  3.4515970e-04]], dtype=float32),\n","             array([ 2.7334734e-04, -1.8405043e-04,  1.0120308e-04, ...,\n","                     8.6085325e-05,  1.3715283e-04, -8.6703437e-07], dtype=float32)),\n","            ((), (), ())))),\n","         (),\n","         ((), (), ()),\n","         (array([[ 6.0939236e-04, -6.4423645e-04, -6.8932719e-04, ...,\n","                  -3.9897865e-04,  2.5841341e-04,  5.9646647e-04],\n","                 [-1.3304035e-04,  2.9796452e-04, -4.0251299e-04, ...,\n","                  -1.4373536e-04,  3.0255024e-04, -2.3298207e-04],\n","                 [-1.5430401e-04,  4.3703345e-04, -1.1130284e-03, ...,\n","                  -3.8600952e-05,  4.5966762e-04, -1.1355604e-04],\n","                 ...,\n","                 [ 2.3915965e-04, -2.1222497e-04,  3.9562778e-04, ...,\n","                   3.4314345e-04, -3.0463800e-04, -5.1199534e-04],\n","                 [ 4.1762739e-04, -6.1627413e-04,  8.7624481e-05, ...,\n","                   1.3334911e-04,  3.1674004e-04, -1.8183839e-04],\n","                 [-5.9693685e-04,  2.1780561e-05, -9.9079334e-04, ...,\n","                  -5.9006334e-04,  1.7320567e-04, -3.6406913e-04]], dtype=float32),\n","          array([-1.0726004e-04, -1.1833757e-05,  1.3794427e-04, ...,\n","                  1.6053069e-04, -9.1469839e-05,  6.0329181e-05], dtype=float32))),\n","        ()),\n","       ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([32.962135, 33.682213, 37.371616, ..., 38.854427, 35.33875 ,\n","              29.180626], dtype=float32),\n","       array([-1.2618901e-04,  8.9752757e-05, -1.7549617e-04, ...,\n","              -2.8014110e-04,  1.9493887e-04, -2.4300600e-04], dtype=float32)),\n","      (),\n","      ((), (((array([[ 2.40848633e-04,  3.91237932e-04,  6.00517495e-04, ...,\n","                  -1.44481004e-04, -8.88519106e-04,  2.32874067e-04],\n","                 [ 1.76484493e-04, -3.28677197e-05, -1.05332874e-04, ...,\n","                   2.18495334e-04,  6.33010000e-04,  5.65440976e-04],\n","                 [-2.49794510e-04,  1.06618588e-03, -6.35769160e-04, ...,\n","                   2.79996784e-05,  2.86025723e-04,  1.96141322e-04],\n","                 ...,\n","                 [ 4.70749888e-04,  9.04935470e-04,  6.52731687e-04, ...,\n","                  -1.54277095e-05, -2.03859599e-04,  3.74065014e-04],\n","                 [ 1.57579896e-04, -5.44396287e-04,  2.80965643e-04, ...,\n","                  -4.22512559e-04, -2.62567482e-04, -1.26492488e-03],\n","                 [ 1.24595623e-04, -1.88576116e-04,  5.91425865e-04, ...,\n","                  -9.83647304e-04, -1.03813210e-04,  7.08293577e-04]], dtype=float32),\n","          array([-1.8540796e-04, -4.6116420e-05, -6.7696863e-05, ...,\n","                  1.1634738e-04, -1.6359212e-04,  1.8576148e-04], dtype=float32)),\n","         ((array([[ 6.7053776e-04,  7.6331769e-04, -1.9996277e-04, ...,\n","                   -9.7747031e-04, -2.5796000e-04, -3.8775528e-04],\n","                  [-3.7140574e-04, -1.4802116e-04,  4.4826261e-04, ...,\n","                    2.2452491e-04,  9.7886438e-04,  1.6908179e-04],\n","                  [ 1.3906253e-04,  3.9093973e-04, -6.7381549e-04, ...,\n","                    2.3805875e-04, -1.0471264e-04,  9.0856316e-05],\n","                  ...,\n","                  [ 9.8987482e-05,  4.6520800e-05,  1.0380712e-03, ...,\n","                    6.9878163e-04, -1.5667704e-04, -1.5810062e-04],\n","                  [-1.1372827e-03,  5.9662245e-05,  5.1317317e-04, ...,\n","                   -3.9829072e-04, -3.3046817e-04,  8.4939686e-04],\n","                  [ 1.3907775e-04,  1.0263787e-03,  1.4517606e-04, ...,\n","                    7.4170949e-04, -3.0485497e-04,  6.7392638e-04]], dtype=float32),\n","           array([ 2.2246757e-04, -2.4376388e-04,  4.1273388e-04, ...,\n","                  -2.8938841e-04, -3.4699476e-04, -2.6670448e-05], dtype=float32)),),\n","         ((array([[-3.76365293e-04,  5.34977298e-04,  1.09476736e-04, ...,\n","                    1.30244895e-04,  3.17033904e-04,  4.13547095e-04],\n","                  [ 1.03219226e-03,  9.23885789e-04, -5.21125621e-04, ...,\n","                   -4.56239301e-04, -8.18487082e-04,  8.03160598e-04],\n","                  [ 1.76983289e-04,  2.18045665e-04, -1.23251215e-04, ...,\n","                    1.56616108e-04, -4.67672158e-04,  1.64088749e-04],\n","                  ...,\n","                  [ 4.33076311e-05,  9.48890374e-05, -9.69107263e-04, ...,\n","                   -2.74306280e-04, -1.41725555e-04,  7.08586304e-05],\n","                  [ 6.56651682e-04,  6.72535680e-04, -3.10556788e-04, ...,\n","                   -1.69744962e-04, -4.55862930e-04, -6.97582320e-04],\n","                  [-6.36886107e-05,  3.39899998e-04,  3.31482355e-04, ...,\n","                   -5.60478657e-04,  3.40689206e-04, -2.33069863e-04]], dtype=float32),\n","           array([ 3.2288248e-05, -1.4159526e-04, -1.9391895e-05, ...,\n","                  -2.9895578e-05,  8.8007655e-05, -2.0805126e-05], dtype=float32)),)),\n","        (),\n","        (array([[ 3.2272670e-04, -1.1836680e-04,  3.9069775e-05, ...,\n","                  6.3512439e-04, -2.6915877e-04, -1.2416736e-04],\n","                [-6.5422704e-04, -6.5429143e-05,  3.7028332e-04, ...,\n","                  2.4551639e-04, -4.9553084e-04, -1.3843182e-04],\n","                [ 5.2893098e-05, -6.5914099e-04, -6.9230678e-04, ...,\n","                 -9.7391702e-04, -5.7645935e-05,  5.9345810e-05],\n","                ...,\n","                [-9.7795750e-04,  9.1507484e-04,  7.5163774e-04, ...,\n","                  2.1974967e-04, -4.6315407e-05, -1.0808606e-05],\n","                [ 5.7364570e-05,  9.5007572e-06, -2.0947275e-04, ...,\n","                 -9.6154347e-04, -3.6487661e-05, -3.0533277e-04],\n","                [ 3.6584199e-04, -8.4189400e-05, -5.0124207e-05, ...,\n","                 -5.9376307e-06, -5.9717899e-04,  5.2334525e-04]], dtype=float32),\n","         array([ 4.0987983e-05, -9.0039910e-05,  2.6817984e-04, ...,\n","                 1.7860872e-04,  3.8798382e-05,  6.4128057e-05], dtype=float32))), ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([43.404766, 32.451645, 35.212673, ..., 41.03516 , 32.295918,\n","              40.778492], dtype=float32),\n","       array([-0.00838077,  0.00323922,  0.00070724, ..., -0.00224101,\n","               0.00397437,  0.00752799], dtype=float32)),\n","      (array([[-2.49023578e-05,  3.05114139e-04,  5.40365836e-05, ...,\n","                2.60564440e-04, -2.44660623e-04,  3.21743079e-04],\n","              [-5.92707947e-04,  5.23478026e-04,  3.16385413e-04, ...,\n","                6.80177065e-04, -5.45618357e-04, -1.34587070e-04],\n","              [-7.07946485e-04,  2.84677604e-04, -4.02061632e-05, ...,\n","               -7.68111568e-05, -9.59627578e-05,  8.65133246e-04],\n","              ...,\n","              [-2.22533854e-04,  3.46814340e-04, -4.23439924e-04, ...,\n","               -1.33673646e-04,  4.38650808e-04, -3.64539737e-04],\n","              [-2.56185886e-04,  2.80157110e-04, -2.67453637e-04, ...,\n","                2.94620259e-04,  1.03755636e-04,  6.15080236e-04],\n","              [ 9.82113706e-05,  6.01431122e-04,  9.87708700e-05, ...,\n","                8.69020805e-05,  2.68182484e-05, -9.66138206e-04]], dtype=float32),\n","       array([-1.9460889e-04, -2.6786275e-04, -3.8324219e-05, ...,\n","              -1.8253500e-04, -9.9693010e-05, -3.7107512e-04], dtype=float32)),\n","      ((), (), ()),\n","      (),\n","      (array([[-6.9728057e-04, -1.5398908e-04,  9.1202551e-04, ...,\n","               -5.9193131e-05, -9.8032178e-04, -1.8543695e-04],\n","              [-2.0064102e-04,  5.7658833e-04, -2.9566791e-04, ...,\n","                2.4934762e-04, -2.5478503e-04,  1.2016719e-03],\n","              [-3.0364061e-04,  2.7701649e-04,  2.9131475e-05, ...,\n","                1.0455367e-04,  1.9933682e-04,  3.1445984e-04],\n","              ...,\n","              [ 9.5466460e-04,  4.5652007e-04, -3.5080581e-04, ...,\n","                1.3052080e-04,  3.4946419e-04, -1.1246263e-05],\n","              [ 3.3033013e-04, -3.6177036e-04,  5.1580917e-04, ...,\n","               -3.6729241e-04, -1.4127331e-04,  6.5164070e-04],\n","              [-2.0411615e-05,  3.8182017e-04,  4.7749741e-04, ...,\n","                1.3978056e-04,  6.7537832e-05, -2.9754010e-04]], dtype=float32),\n","       array([ 0.00022498, -0.00028898,  0.00025087, ...,  0.0003228 ,\n","              -0.00028461, -0.00020158], dtype=float32)),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([24.100801, 23.241558, 23.465302, ..., 26.858469, 26.697847,\n","              25.494972], dtype=float32),\n","       array([-0.00059024,  0.00025571,  0.0001835 , ...,  0.00024984,\n","               0.00062651,  0.00017387], dtype=float32)),\n","      ((),\n","       ((),\n","        (((), (((array([[-6.3081237e-04,  5.3716020e-04, -2.6521087e-04, ...,\n","                     -7.0814148e-04,  3.5875579e-04,  1.0547517e-04],\n","                    [ 4.7492766e-05, -3.3656627e-04,  3.3182444e-04, ...,\n","                     -3.2390203e-04, -5.8206944e-05, -2.8998719e-04],\n","                    [-9.2797294e-05, -5.8914622e-04,  9.6110871e-06, ...,\n","                     -3.4542038e-04,  1.3779261e-04, -1.3671294e-04],\n","                    ...,\n","                    [-2.5119606e-04,  1.4903853e-04, -1.0695050e-03, ...,\n","                      4.1271691e-04, -9.2966467e-05,  9.1911567e-04],\n","                    [ 8.3633291e-04, -1.0860865e-03,  1.8169268e-04, ...,\n","                     -1.0918509e-03,  7.7419850e-04, -4.7316487e-04],\n","                    [-3.3056471e-04,  9.9459168e-05, -5.6943383e-05, ...,\n","                      6.7701045e-04, -5.4551760e-04, -1.0767155e-03]], dtype=float32),\n","             array([-2.8268498e-04,  2.2540537e-04, -7.8904571e-04, ...,\n","                     6.9108733e-05, -5.4621568e-04,  1.4977253e-04], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[-9.65842570e-04,  5.53148566e-04, -1.54867826e-04, ...,\n","                     -6.42235798e-04,  1.62274853e-04,  2.42566777e-04],\n","                    [-1.08520710e-03,  8.19937050e-05,  3.24418856e-04, ...,\n","                     -3.78924393e-04,  1.54243564e-04, -6.84584491e-04],\n","                    [-1.34743677e-04, -1.05324457e-03,  3.01364838e-04, ...,\n","                     -1.03265687e-04,  2.01348419e-04,  4.19369084e-04],\n","                    ...,\n","                    [ 7.35590118e-04,  4.32439789e-04, -7.15718837e-04, ...,\n","                     -3.48186149e-04,  1.15018134e-04,  2.38689201e-04],\n","                    [-4.56294365e-05, -4.21075179e-04, -5.79830026e-04, ...,\n","                     -6.11158088e-04,  7.90136401e-04, -7.79223337e-04],\n","                    [ 3.44493659e-04,  2.34370775e-04,  6.03105545e-05, ...,\n","                      6.06559799e-04, -1.25002748e-06, -1.96235473e-04]], dtype=float32),\n","             array([ 5.8568861e-05, -3.4975532e-05,  2.5956787e-04, ...,\n","                     3.5377071e-04,  2.0791097e-04, -1.8495474e-04], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[ 1.1213715e-03, -8.5139996e-04, -1.1271482e-04, ...,\n","                      3.9399692e-04,  6.6564581e-04,  1.6451288e-04],\n","                    [ 3.3034943e-04,  1.6099680e-06,  8.5477048e-05, ...,\n","                     -8.4187923e-05,  4.7738249e-06, -1.4681117e-04],\n","                    [-6.5046275e-04, -2.8924094e-04, -2.8990788e-04, ...,\n","                     -7.1199625e-05, -1.8924405e-04,  1.3306594e-04],\n","                    ...,\n","                    [ 1.2440166e-04,  8.0427714e-04,  4.6415086e-04, ...,\n","                      1.2217369e-03, -2.2442078e-05, -3.2459579e-05],\n","                    [ 2.3060113e-04, -2.5813826e-04,  1.7235641e-04, ...,\n","                     -2.0273046e-04,  5.4986548e-04, -6.2927499e-04],\n","                    [-2.6753396e-04, -1.4796147e-04, -1.5441798e-04, ...,\n","                      5.0313747e-04,  4.1388863e-04, -7.8875841e-05]], dtype=float32),\n","             array([ 8.7556778e-05, -3.5270385e-04,  6.8494170e-05, ...,\n","                    -1.0414499e-04,  5.7217890e-06,  4.8879316e-05], dtype=float32)),\n","            ((), (), ())))),\n","         (),\n","         ((), (), ()),\n","         (array([[-2.9806572e-04, -1.9154622e-04,  2.0637360e-04, ...,\n","                   2.9280623e-05, -1.2881309e-04,  2.6983400e-05],\n","                 [ 5.0714461e-04, -5.9929944e-04, -2.5615953e-05, ...,\n","                  -6.9816830e-04,  1.0167210e-03,  8.4102311e-04],\n","                 [ 5.5506668e-04, -2.9061010e-04,  7.5494859e-04, ...,\n","                  -1.5574814e-04,  3.2180626e-04,  1.0845118e-03],\n","                 ...,\n","                 [-3.4966761e-05, -3.8362044e-04,  2.9349123e-04, ...,\n","                  -5.0203910e-04, -3.9054037e-04,  1.0474048e-04],\n","                 [-2.5996525e-04,  3.0632326e-04, -6.2626111e-04, ...,\n","                  -2.1689411e-04,  4.8785971e-04,  4.2506359e-05],\n","                 [ 1.7795811e-04,  8.7476615e-04,  3.7046536e-04, ...,\n","                   1.5550104e-04,  9.5489994e-04, -3.1480397e-04]], dtype=float32),\n","          array([-8.0975180e-05, -1.6588147e-05,  2.5008351e-04, ...,\n","                  3.6657645e-04, -7.0206130e-05, -4.5262503e-05], dtype=float32))),\n","        ()),\n","       ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([33.473694, 38.033276, 34.017498, ..., 36.435352, 32.368355,\n","              29.591436], dtype=float32),\n","       array([ 3.7834144e-04, -5.1072006e-05,  2.5769259e-04, ...,\n","              -1.9201488e-04, -7.6976634e-05, -2.0807332e-04], dtype=float32)),\n","      (),\n","      ((), (((array([[-2.0333480e-04,  3.5288405e-05,  6.1436452e-04, ...,\n","                  -3.1081293e-04,  5.3271675e-04, -2.2776301e-04],\n","                 [ 1.5003869e-04, -2.4098623e-05,  4.4034095e-04, ...,\n","                   3.2970440e-04, -4.0633947e-04,  4.5295886e-04],\n","                 [-1.6040583e-05,  1.0814344e-03,  5.3932751e-04, ...,\n","                  -5.7865202e-04, -1.0140367e-03,  4.4210337e-04],\n","                 ...,\n","                 [ 1.0136795e-04, -9.2392572e-04,  6.5981713e-04, ...,\n","                  -1.2151604e-04, -8.4981997e-04, -8.4481982e-04],\n","                 [-2.9921077e-05, -2.0164209e-04, -1.6544100e-04, ...,\n","                  -1.4043476e-04, -1.7713086e-04, -1.8400968e-03],\n","                 [-4.3528715e-05,  1.0856190e-03,  4.8063273e-04, ...,\n","                   5.7630450e-04, -2.9514483e-04, -8.5334561e-04]], dtype=float32),\n","          array([-2.3082473e-05,  1.7025610e-04,  3.3550172e-05, ...,\n","                  2.3292989e-04, -8.8557499e-05,  5.7075504e-05], dtype=float32)),\n","         ((array([[ 1.6223016e-04,  9.1843324e-04, -5.2405994e-05, ...,\n","                   -4.9283745e-04, -9.1003266e-04,  4.2166532e-04],\n","                  [-6.5762294e-04,  1.3240919e-03, -2.5095357e-04, ...,\n","                   -6.7857618e-04, -6.9535093e-04, -1.0748972e-03],\n","                  [ 2.2254794e-04,  5.6550518e-04, -2.5325525e-04, ...,\n","                    1.6012079e-04, -6.8526002e-05,  3.1954274e-04],\n","                  ...,\n","                  [-8.5724203e-04,  1.0243553e-04, -6.5549713e-04, ...,\n","                   -9.7829194e-05, -1.9738627e-05, -3.5057217e-04],\n","                  [-3.2161493e-04, -4.7108420e-04,  8.6443184e-04, ...,\n","                   -5.8776088e-04,  9.9527102e-04,  5.1328377e-04],\n","                  [ 2.0462249e-03, -7.3939946e-04, -7.6414511e-04, ...,\n","                    1.4393931e-03, -2.6952146e-04, -5.9983705e-04]], dtype=float32),\n","           array([ 3.6205515e-06, -1.2989414e-04, -3.0502581e-04, ...,\n","                  -1.4278670e-04, -1.5954168e-04,  2.2894159e-04], dtype=float32)),),\n","         ((array([[ 1.61416086e-04, -3.19579762e-04,  2.85720162e-04, ...,\n","                    7.74007582e-04,  8.05685922e-05, -6.89081615e-04],\n","                  [ 1.39711599e-04, -8.07319477e-04,  9.53182636e-04, ...,\n","                   -2.17131106e-04, -1.00315176e-03,  8.25955940e-04],\n","                  [ 3.85121675e-04,  2.30686259e-04, -1.47962812e-04, ...,\n","                   -2.22763454e-04, -2.09041245e-04,  7.32300077e-06],\n","                  ...,\n","                  [-5.12645813e-04, -5.57653490e-04, -1.28614518e-03, ...,\n","                    1.68623403e-04, -6.75717747e-05, -1.05883475e-04],\n","                  [ 2.78576801e-04, -3.99299315e-04,  2.12386542e-04, ...,\n","                    2.27691678e-04, -9.50011599e-04, -3.85453313e-04],\n","                  [ 2.02174546e-04,  3.06900736e-04,  3.14365403e-04, ...,\n","                    4.22776153e-04, -6.81680103e-04, -1.21370635e-04]], dtype=float32),\n","           array([-3.0842208e-04, -1.6140464e-05, -8.8436187e-05, ...,\n","                  -9.9335584e-06,  1.9472861e-04, -9.5469506e-05], dtype=float32)),)),\n","        (),\n","        (array([[ 1.8605105e-04,  4.4909539e-04, -4.8060043e-04, ...,\n","                  1.6590783e-04, -6.3875086e-05, -2.5659995e-04],\n","                [-2.9511156e-04,  7.2396419e-05, -1.4551599e-04, ...,\n","                  6.4530747e-04,  3.0823564e-04,  3.0197846e-04],\n","                [-5.7197572e-04,  5.4059303e-05,  4.9826165e-04, ...,\n","                  1.2151337e-03,  8.8153593e-04,  8.2635600e-04],\n","                ...,\n","                [-4.6077129e-04,  2.4119439e-04,  1.1833304e-04, ...,\n","                 -8.9524283e-06, -8.7501372e-05, -5.4545893e-04],\n","                [-8.0175145e-04, -1.0490407e-03,  3.4331670e-04, ...,\n","                 -8.1683917e-04,  3.4321929e-04,  3.4729415e-04],\n","                [ 2.0694277e-04,  8.3909550e-04, -8.7526161e-04, ...,\n","                 -1.2257358e-04, -2.6392311e-04, -7.3279289e-04]], dtype=float32),\n","         array([-0.00023603, -0.00013752,  0.00019173, ...,  0.00031368,\n","                -0.00017833,  0.00013501], dtype=float32))), ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([49.65025 , 36.382725, 37.7449  , ..., 43.311428, 35.772926,\n","              45.91778 ], dtype=float32),\n","       array([-0.00421857,  0.00117495,  0.00187945, ...,  0.0003378 ,\n","               0.00305881,  0.00316152], dtype=float32)),\n","      (array([[-3.16245336e-04,  4.59237490e-04, -1.08626431e-04, ...,\n","                6.42700936e-04, -4.43487748e-04,  2.45134055e-04],\n","              [-8.38938693e-04,  3.12759716e-04, -4.15371869e-05, ...,\n","               -5.11566876e-04, -1.21783014e-04,  7.30592947e-05],\n","              [ 6.75311079e-04, -1.82123811e-04, -5.40124194e-04, ...,\n","               -2.09182632e-04, -5.37234184e-04,  1.75023306e-04],\n","              ...,\n","              [-2.69602693e-04,  7.22777739e-04, -6.04090339e-04, ...,\n","                1.63147488e-04,  1.39024065e-04,  4.02513397e-04],\n","              [ 4.92173887e-04, -8.63095687e-04,  5.61890134e-04, ...,\n","               -2.83185364e-04,  3.60975013e-04,  2.11008446e-05],\n","              [ 5.10636601e-04,  1.73559078e-04,  1.25458959e-04, ...,\n","                1.57329632e-04, -3.07887531e-04,  1.31510445e-04]], dtype=float32),\n","       array([-0.00017219, -0.00020827, -0.00014998, ..., -0.00025824,\n","              -0.00022327, -0.00021303], dtype=float32)),\n","      ((), (), ()),\n","      (),\n","      (array([[ 1.9385938e-04, -5.1879307e-04, -3.0251569e-05, ...,\n","                1.8183270e-04, -5.3268176e-04, -7.1600436e-05],\n","              [-4.1089128e-04, -2.7496583e-04,  7.1638194e-04, ...,\n","               -7.9817901e-04, -6.4723525e-04, -4.0564465e-04],\n","              [ 3.4336761e-05, -6.8837973e-05,  3.7499337e-04, ...,\n","                8.5161613e-05,  6.3212996e-04, -8.8624179e-04],\n","              ...,\n","              [-1.3445862e-04, -3.0630908e-04, -5.0410349e-04, ...,\n","                2.6085725e-04, -8.8609188e-05,  8.4811600e-04],\n","              [ 4.2543357e-04, -3.0292437e-04,  2.1181330e-04, ...,\n","                6.3684158e-04, -6.2127132e-04, -1.0300137e-03],\n","              [-5.8099139e-04,  2.1684957e-04,  2.0593328e-04, ...,\n","               -3.9433967e-06,  2.0224910e-04, -4.3778936e-04]], dtype=float32),\n","       array([ 0.00012507, -0.00026121,  0.00042177, ...,  0.00025099,\n","              -0.00028355, -0.00012039], dtype=float32)),\n","      ()))),\n","   ()),\n","  (((),\n","    ((),\n","     ((array([12.257917 , 13.803502 , 11.657565 , ..., 13.611211 , 14.4332905,\n","              11.248412 ], dtype=float32),\n","       array([ 1.7210263e-03,  5.5674747e-05,  4.1065592e-04, ...,\n","               5.8492232e-04, -4.0599590e-04, -7.9247280e-04], dtype=float32)),\n","      ((),\n","       ((),\n","        (((), (((array([[ 5.4224598e-04,  1.5834977e-03,  7.8646222e-04, ...,\n","                     -6.0450041e-04,  7.0957071e-04, -4.3461486e-04],\n","                    [-8.6372113e-04, -1.6070801e-03, -2.0270064e-03, ...,\n","                      6.6899718e-04,  1.2105798e-03,  3.1111651e-04],\n","                    [-8.2303584e-04,  2.1290593e-03, -2.5450578e-03, ...,\n","                      9.0269366e-04,  8.4327289e-04,  4.7788527e-04],\n","                    ...,\n","                    [ 3.6869754e-04, -1.6585771e-03, -3.9118304e-04, ...,\n","                      8.1556599e-04, -1.2148699e-03, -2.5728880e-04],\n","                    [-2.5726431e-05,  8.3054128e-06, -6.5881317e-04, ...,\n","                     -1.2647860e-04,  4.0197061e-04, -1.1220530e-04],\n","                    [-6.4471376e-04, -1.0294451e-03, -2.7059778e-04, ...,\n","                      3.8380944e-04,  4.5579605e-04,  7.6313555e-04]], dtype=float32),\n","             array([-1.0984585e-04,  1.4402825e-04, -5.0859165e-04, ...,\n","                     3.2742508e-04,  1.0514637e-05,  2.2199764e-04], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[-8.8539811e-05,  2.6066673e-03, -1.0599779e-03, ...,\n","                      2.7882229e-04,  7.1414886e-04,  7.8822934e-04],\n","                    [ 1.4262641e-03, -2.0354846e-03,  3.8212709e-04, ...,\n","                      7.0961699e-04, -5.8371411e-04,  5.5760535e-04],\n","                    [ 8.5399015e-04,  1.4881040e-03, -2.0657289e-03, ...,\n","                      1.9827751e-04, -9.5943460e-04, -7.2677556e-04],\n","                    ...,\n","                    [ 2.9360992e-04, -2.4838257e-04, -2.3642198e-04, ...,\n","                     -3.6941914e-04,  2.3566422e-04,  7.4241427e-04],\n","                    [ 9.7598607e-04,  1.9018227e-04, -1.5528607e-03, ...,\n","                     -1.3449257e-04, -6.4952741e-04, -1.2552411e-03],\n","                    [-9.7025314e-04, -2.0964369e-03, -2.6112306e-04, ...,\n","                      3.9391374e-04,  1.7261036e-05,  1.5032378e-03]], dtype=float32),\n","             array([-4.5976245e-05,  9.9804878e-05,  4.0433113e-04, ...,\n","                     2.5986117e-05,  3.4175479e-04,  2.9991704e-04], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[ 2.22052506e-04, -9.63209488e-04, -5.25782874e-04, ...,\n","                     -3.39349877e-04, -4.39125550e-04, -1.10384426e-04],\n","                    [-1.91413637e-04,  5.91859571e-04, -1.40286458e-04, ...,\n","                      6.01850043e-04, -1.50130130e-04,  2.78175139e-04],\n","                    [-5.03792020e-04,  8.82100954e-04,  4.68105776e-04, ...,\n","                      1.66712533e-04,  2.52411148e-04,  5.88535855e-04],\n","                    ...,\n","                    [ 4.03014565e-04,  2.23617841e-04, -5.03102667e-04, ...,\n","                      1.85462104e-05,  1.14007620e-04,  3.04938265e-04],\n","                    [-1.39969445e-04,  4.39587457e-04,  7.80663686e-04, ...,\n","                      5.92898286e-04,  7.90434948e-04, -4.58822142e-05],\n","                    [-3.43795051e-04, -3.59958009e-04,  4.27016144e-04, ...,\n","                      2.63658148e-04,  9.01883293e-04,  1.15126408e-04]], dtype=float32),\n","             array([ 1.1636445e-04, -7.0630565e-05, -1.4578798e-05, ...,\n","                     2.6265636e-05,  4.3365981e-05,  8.6353451e-05], dtype=float32)),\n","            ((), (), ())))),\n","         (),\n","         ((), (), ()),\n","         (array([[-4.0357851e-04,  8.2190381e-05,  2.5413849e-04, ...,\n","                  -5.2808449e-05,  3.1512722e-05, -3.2942859e-05],\n","                 [ 3.4810841e-05, -3.2847180e-04, -1.0942692e-04, ...,\n","                  -1.9452897e-04,  1.3700755e-04,  5.6684169e-04],\n","                 [ 3.2029286e-04, -3.3619159e-04, -9.3283224e-06, ...,\n","                   1.6554238e-04, -8.4564927e-05,  3.0961957e-05],\n","                 ...,\n","                 [-2.2494508e-04,  5.7906464e-06, -4.7692738e-04, ...,\n","                   7.0483919e-05,  3.5792077e-04, -2.1393465e-05],\n","                 [-5.9387246e-05, -3.5693229e-04, -2.6513857e-04, ...,\n","                   3.6162211e-04, -9.2544840e-05, -4.9655435e-05],\n","                 [ 6.3284278e-05, -8.2766310e-05,  9.3892428e-05, ...,\n","                  -9.3604096e-05,  3.5746186e-04, -2.3742932e-06]], dtype=float32),\n","          array([-1.7298094e-04, -9.1816386e-05,  1.5238310e-04, ...,\n","                  1.3456921e-04,  1.2621414e-04, -5.6411859e-06], dtype=float32))),\n","        ()),\n","       ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([29.174044, 24.393362, 26.36628 , ..., 25.530327, 26.251175,\n","              21.72299 ], dtype=float32),\n","       array([ 4.7743692e-05, -3.6738558e-05,  9.7822725e-05, ...,\n","              -1.0868642e-04, -3.0023188e-04, -1.6403514e-04], dtype=float32)),\n","      (),\n","      ((), (((array([[-3.24723835e-04, -1.68654296e-04, -1.07442099e-03, ...,\n","                   1.35239054e-04, -1.35051407e-04,  9.53159528e-04],\n","                 [ 1.96144465e-04,  3.13790719e-04,  4.11479996e-04, ...,\n","                   1.35934236e-03,  3.76634853e-04,  6.68696361e-04],\n","                 [-2.71165685e-04, -7.97942164e-04,  1.33597816e-04, ...,\n","                   1.44457226e-05,  6.20037550e-04,  5.30940073e-04],\n","                 ...,\n","                 [ 5.47067961e-04,  4.43779980e-04,  1.70805884e-04, ...,\n","                   1.60784461e-04,  1.97699810e-05, -1.78085058e-04],\n","                 [ 1.09161854e-04, -2.30359146e-05, -8.75763013e-04, ...,\n","                  -2.94599682e-04, -1.21328281e-03, -3.22645006e-04],\n","                 [-3.42633954e-04, -8.49065254e-04,  3.52772389e-04, ...,\n","                   1.11944869e-03,  4.68079379e-04,  5.46326744e-04]], dtype=float32),\n","          array([-4.7888454e-05,  1.7187731e-04,  7.4867124e-04, ...,\n","                 -1.9099271e-04, -2.1119138e-04, -4.3064437e-04], dtype=float32)),\n","         ((array([[ 5.9134501e-04, -4.5603761e-04, -6.8353242e-05, ...,\n","                   -8.1594911e-04, -6.6387490e-04,  8.5665670e-04],\n","                  [ 1.0682491e-03, -6.7269430e-04, -6.2339252e-04, ...,\n","                    1.7048493e-04,  5.6462735e-04, -8.4948691e-04],\n","                  [ 1.1353758e-03, -6.7690708e-05, -3.2356661e-04, ...,\n","                   -5.9274980e-04, -1.3942050e-04, -2.2288394e-04],\n","                  ...,\n","                  [-1.0829545e-03,  9.1623742e-04, -1.6607143e-04, ...,\n","                   -2.6985415e-05, -2.8520747e-04, -1.2178086e-04],\n","                  [ 4.9321021e-05,  7.0944830e-04, -1.9194186e-04, ...,\n","                   -4.4635349e-04,  9.6453342e-04, -4.3109537e-04],\n","                  [-9.7005616e-04, -1.0159230e-03, -2.3685575e-04, ...,\n","                   -2.7841813e-04, -1.4393379e-03, -3.1993698e-04]], dtype=float32),\n","           array([ 2.6712462e-04, -2.0817308e-04,  5.7503692e-04, ...,\n","                  -1.3085313e-05, -7.4476811e-05, -3.5462249e-04], dtype=float32)),),\n","         ((array([[ 3.1561602e-04, -3.9927135e-04, -6.0264640e-05, ...,\n","                   -2.8513925e-04,  6.9047269e-06, -2.1178849e-04],\n","                  [-1.4089516e-03, -1.1973451e-03,  9.2289680e-05, ...,\n","                   -5.6787778e-04,  5.6926167e-04, -7.6816633e-04],\n","                  [ 9.6055999e-05, -1.1942562e-04,  2.2005485e-04, ...,\n","                    1.7801563e-04, -1.6369559e-04,  1.2885647e-04],\n","                  ...,\n","                  [-7.9876598e-04,  6.4870558e-04,  5.1448638e-05, ...,\n","                   -1.4707012e-03,  4.1508931e-04, -5.2305235e-04],\n","                  [-1.1084215e-03, -4.8864668e-04, -3.9870027e-04, ...,\n","                    4.1991286e-04, -9.5220917e-04,  6.8766496e-04],\n","                  [ 3.6123405e-05, -8.6336269e-04,  1.5379385e-06, ...,\n","                   -3.7839284e-04, -3.2075905e-04,  1.3585578e-04]], dtype=float32),\n","           array([ 2.4563848e-05, -1.5738969e-04, -2.1878723e-05, ...,\n","                   2.0783902e-04, -3.3467953e-05,  1.8562055e-04], dtype=float32)),)),\n","        (),\n","        (array([[-9.5273077e-05,  2.8270228e-05, -5.0107052e-04, ...,\n","                  6.0823467e-04,  3.8311406e-04, -1.0139835e-03],\n","                [ 1.8976421e-04, -3.6743161e-04,  1.1354777e-04, ...,\n","                  2.1268002e-04,  5.5170601e-04, -1.6835122e-04],\n","                [ 1.0876931e-03,  1.5127742e-04,  1.2043151e-04, ...,\n","                 -4.9039582e-04, -3.4397517e-04, -7.9544069e-04],\n","                ...,\n","                [ 1.6855577e-04,  2.0505962e-04,  2.5704331e-04, ...,\n","                 -5.7559414e-04, -5.2894861e-04,  2.3671266e-04],\n","                [-5.3056178e-04,  5.3933315e-04, -5.1768770e-04, ...,\n","                  2.1520574e-04, -1.5905108e-04, -2.0962034e-04],\n","                [-7.3016185e-05, -3.0274357e-04,  4.9791351e-04, ...,\n","                  2.1644084e-04,  3.9030021e-04, -7.4902945e-04]], dtype=float32),\n","         array([-1.84661214e-04, -2.21554670e-04,  6.80204466e-05, ...,\n","                 2.58682732e-04, -8.10266793e-05,  1.02047314e-04], dtype=float32))), ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([49.74295 , 41.72023 , 47.380295, ..., 43.96252 , 41.160122,\n","              51.336563], dtype=float32),\n","       array([-0.00813712,  0.00128333,  0.00182296, ...,  0.00079721,\n","               0.00280662,  0.0037483 ], dtype=float32)),\n","      (array([[ 6.14974182e-04,  4.04020866e-05, -1.61048607e-04, ...,\n","                2.68221542e-04, -4.55199770e-04,  6.15576748e-04],\n","              [-2.50311248e-04, -1.08902204e-04, -2.20029600e-04, ...,\n","               -6.37460093e-04,  1.01482612e-04, -4.06629842e-04],\n","              [ 1.39570795e-04, -2.61553796e-04,  2.77892570e-04, ...,\n","                3.64282023e-04, -2.97136448e-05,  5.96397149e-04],\n","              ...,\n","              [ 1.35258844e-04,  4.64393059e-04, -4.38057643e-04, ...,\n","                3.76670010e-04, -2.54353625e-04, -8.68873321e-05],\n","              [-9.19627375e-04, -2.85332120e-04, -3.30303374e-05, ...,\n","                2.98238068e-04, -3.70562135e-04, -1.93337808e-04],\n","              [ 3.43492575e-04, -4.15336370e-04, -3.07645329e-04, ...,\n","               -4.69973747e-05,  3.75064614e-04, -5.87708491e-04]], dtype=float32),\n","       array([-0.00010872, -0.00021807, -0.00026602, ..., -0.00033584,\n","              -0.00013251, -0.00021067], dtype=float32)),\n","      ((), (), ()),\n","      (),\n","      (array([[-4.25396196e-04, -1.24882348e-03, -2.57539014e-05, ...,\n","                3.06803122e-04, -9.05163412e-04,  1.21356672e-04],\n","              [-1.75216352e-04,  4.71273583e-04, -1.32851943e-04, ...,\n","                5.01035189e-04,  1.09895249e-03, -6.42601401e-04],\n","              [-4.14501585e-04, -5.08909325e-05,  2.29051111e-05, ...,\n","                1.70716608e-04,  1.04989915e-03, -1.57524846e-05],\n","              ...,\n","              [-1.78306509e-04, -7.92282081e-05,  4.55497677e-04, ...,\n","                3.22547276e-05,  8.38483917e-04,  1.44377933e-04],\n","              [-3.28086840e-04, -4.78912116e-04, -3.71525530e-04, ...,\n","               -1.99182934e-04,  6.35698321e-04, -6.88720436e-04],\n","              [ 4.38584480e-04, -1.19491146e-04, -4.41714190e-04, ...,\n","                7.82078190e-04,  6.72052149e-04, -3.50740765e-06]], dtype=float32),\n","       array([ 4.0298604e-05, -3.9617367e-05,  2.3531655e-04, ...,\n","               2.6399741e-04, -3.8635480e-04, -2.2709252e-04], dtype=float32)),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([24.067898, 24.463842, 24.93064 , ..., 24.311825, 29.93596 ,\n","              23.747051], dtype=float32),\n","       array([ 1.1951851e-04,  2.2809544e-04, -9.9192621e-06, ...,\n","              -1.0882222e-04, -7.7869503e-05, -1.6176696e-04], dtype=float32)),\n","      ((),\n","       ((),\n","        (((), (((array([[ 1.3401081e-03,  2.4956194e-04,  3.0726197e-04, ...,\n","                     -9.2942297e-04, -6.1775476e-04,  4.0824065e-04],\n","                    [-1.1768404e-03, -2.0657622e-04,  3.2524990e-06, ...,\n","                      7.8251841e-04,  4.8062578e-04, -3.2890199e-05],\n","                    [-6.4378127e-04, -1.2508576e-03, -1.2538311e-04, ...,\n","                      9.4420562e-04, -4.5705866e-04, -1.0232510e-03],\n","                    ...,\n","                    [-3.4940345e-04, -3.2624986e-04, -9.3882106e-04, ...,\n","                     -1.9501763e-03, -3.9383062e-04,  1.1243374e-03],\n","                    [-3.5890128e-04, -3.5568366e-05, -2.3805804e-03, ...,\n","                      3.9210301e-04, -2.5423925e-04,  2.9659236e-04],\n","                    [ 2.4167185e-04,  9.9893124e-04,  6.8287668e-04, ...,\n","                      5.6180943e-06, -1.4333747e-03, -7.3953980e-04]], dtype=float32),\n","             array([-8.7528559e-04, -1.3112940e-04,  4.7796533e-05, ...,\n","                     3.3904616e-05, -1.7323537e-04,  3.9040190e-04], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[-9.1964798e-04,  1.6685303e-04,  1.7466760e-04, ...,\n","                      8.1812486e-04,  2.4825225e-05, -7.5361175e-05],\n","                    [-5.0158432e-04, -6.8053970e-04,  3.1962144e-04, ...,\n","                      3.3970171e-04,  1.9225864e-04,  1.7475740e-04],\n","                    [ 5.7953985e-05, -1.3019162e-03,  4.1033103e-04, ...,\n","                      2.6295055e-04, -1.0407566e-03,  1.9438434e-05],\n","                    ...,\n","                    [-9.7472814e-04, -6.3422689e-04, -6.6308741e-05, ...,\n","                     -1.1454081e-03,  8.1859319e-04, -1.1697372e-03],\n","                    [ 2.9510036e-04,  9.7229559e-04, -5.2453700e-04, ...,\n","                      1.1779595e-04, -2.3911652e-04,  5.7761132e-04],\n","                    [-9.3378032e-05, -7.7563652e-04,  3.6713752e-04, ...,\n","                     -1.0930051e-04, -8.5474644e-04,  4.8600681e-04]], dtype=float32),\n","             array([-5.6909124e-05,  5.9053535e-04,  1.3233158e-04, ...,\n","                    -8.1265141e-05,  2.7669530e-04, -6.5486915e-05], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[ 3.4368667e-04,  4.1804931e-05, -4.5823667e-04, ...,\n","                     -4.2650977e-04,  3.9083528e-04,  6.3343439e-04],\n","                    [-1.4914642e-04,  2.7553964e-04,  6.6852866e-04, ...,\n","                     -1.6500620e-05, -3.1504669e-04, -6.7653433e-05],\n","                    [ 4.3988461e-04, -4.5583246e-04, -6.1595038e-04, ...,\n","                     -1.3499447e-03, -4.1046719e-06, -5.7197205e-04],\n","                    ...,\n","                    [ 2.6041665e-04,  5.3079292e-04,  1.1580629e-04, ...,\n","                     -1.1099315e-03, -4.6147904e-04, -2.6049901e-04],\n","                    [ 2.9662118e-05, -1.3333722e-03,  6.1666087e-04, ...,\n","                     -2.7387472e-05, -1.5442334e-05,  5.6949502e-04],\n","                    [-2.6196302e-04, -3.3926443e-04, -5.4197444e-04, ...,\n","                      7.4030377e-04, -1.6793745e-04,  1.0517183e-03]], dtype=float32),\n","             array([-1.0785252e-04, -1.9656557e-04, -3.7130681e-05, ...,\n","                    -1.9294053e-05, -4.9866510e-05, -1.3907727e-04], dtype=float32)),\n","            ((), (), ())))),\n","         (),\n","         ((), (), ()),\n","         (array([[ 7.84252072e-04, -1.46511113e-04,  2.54288898e-05, ...,\n","                  -4.21324921e-06,  3.53393116e-04,  1.28579966e-04],\n","                 [ 7.28660147e-04,  3.40340514e-04,  3.62334918e-04, ...,\n","                  -6.73226663e-04,  7.44893396e-06, -1.74050088e-04],\n","                 [ 7.83132331e-04,  7.77001143e-04, -1.12820308e-04, ...,\n","                  -1.11398892e-03,  6.55208540e-04,  1.19864293e-04],\n","                 ...,\n","                 [ 8.81184707e-04, -7.93421641e-04, -4.19859134e-04, ...,\n","                  -3.93786322e-04, -5.55121689e-04,  4.19888063e-04],\n","                 [ 4.81800816e-04, -1.57236675e-04, -1.36516825e-03, ...,\n","                   4.86634672e-05, -2.33378611e-04, -7.03915139e-04],\n","                 [-3.41180596e-04,  4.94836713e-05, -5.09770762e-04, ...,\n","                   1.13928596e-04, -1.50238851e-03, -1.04936911e-03]], dtype=float32),\n","          array([-1.2966366e-04, -1.7647448e-04,  2.0746686e-04, ...,\n","                  2.5649616e-04,  5.2699846e-07,  9.9376382e-05], dtype=float32))),\n","        ()),\n","       ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([25.160736, 26.612818, 23.47876 , ..., 20.231026, 23.498367,\n","              17.000774], dtype=float32),\n","       array([ 2.6102147e-05,  1.4448183e-04, -2.4892788e-04, ...,\n","              -4.9860083e-04, -8.5446234e-05,  1.0175922e-05], dtype=float32)),\n","      (),\n","      ((), (((array([[ 1.5905306e-04,  7.1916125e-05, -1.1138951e-03, ...,\n","                  -8.6008379e-04,  3.8072863e-04,  2.2398231e-04],\n","                 [-1.8046476e-04, -1.6118311e-04, -3.0161548e-04, ...,\n","                  -3.2953927e-04, -8.0109353e-04, -2.7415264e-04],\n","                 [-8.7390811e-04,  1.6487286e-03, -4.1982348e-04, ...,\n","                   1.4574132e-04, -4.4586265e-04,  1.9610819e-04],\n","                 ...,\n","                 [-2.4596398e-04,  1.2222332e-03, -1.2664120e-03, ...,\n","                   8.0826704e-04, -1.9807048e-04,  1.4371097e-04],\n","                 [-2.1376627e-04,  2.4611232e-05,  3.6933125e-04, ...,\n","                   7.0731941e-04,  7.7592900e-05,  7.2204811e-04],\n","                 [-4.1718165e-05, -5.0214358e-04,  1.4120551e-04, ...,\n","                  -6.7586440e-04,  5.6812353e-04, -2.0249156e-04]], dtype=float32),\n","          array([ 0.00028009, -0.00021958,  0.00049121, ..., -0.00020214,\n","                  0.00031906, -0.00029588], dtype=float32)),\n","         ((array([[-6.0727296e-04,  6.5564523e-06,  1.0769475e-03, ...,\n","                   -1.5402302e-03,  1.7850474e-04,  7.8403216e-05],\n","                  [-2.4301361e-04,  7.5262343e-04,  5.7251845e-04, ...,\n","                    4.0739353e-04,  1.6715002e-04,  5.2357904e-05],\n","                  [ 1.4531505e-04,  6.7455330e-05, -2.1372371e-05, ...,\n","                    5.6004227e-04,  7.3772375e-05,  2.8195800e-04],\n","                  ...,\n","                  [ 3.6187699e-05,  7.3064509e-04, -2.6826988e-04, ...,\n","                   -2.6826755e-04,  4.2698888e-05, -2.2242390e-04],\n","                  [ 1.1481734e-05, -9.7529351e-04,  6.4232241e-04, ...,\n","                   -4.0099857e-04, -1.6421331e-04,  1.0467245e-03],\n","                  [ 6.7765330e-04, -1.4986569e-03,  1.7641226e-04, ...,\n","                   -2.2398834e-03, -2.4318052e-04,  8.0162165e-05]], dtype=float32),\n","           array([ 3.9591402e-04, -5.8158239e-05, -2.0523033e-04, ...,\n","                  -1.0956154e-04,  1.3700624e-04,  1.7519935e-04], dtype=float32)),),\n","         ((array([[ 7.12094014e-04, -5.95729107e-05, -1.11275550e-03, ...,\n","                    2.12768136e-04,  4.55667236e-04,  2.64646915e-05],\n","                  [-2.36003092e-04, -1.04406623e-04, -2.71992642e-04, ...,\n","                    1.42015444e-04, -2.41936810e-04, -1.17064737e-04],\n","                  [-7.84665026e-05, -2.28327233e-04, -1.21841585e-04, ...,\n","                   -4.38348972e-04, -1.74307701e-04, -1.59901829e-04],\n","                  ...,\n","                  [-8.99782928e-04,  4.32715693e-04, -5.75106696e-06, ...,\n","                   -6.93478447e-04, -4.51138505e-04,  2.84451817e-04],\n","                  [ 1.58166557e-04,  9.91894281e-04,  6.42707921e-04, ...,\n","                   -1.26365003e-05, -1.99273389e-04,  1.80230214e-04],\n","                  [-3.73001501e-04, -3.86774394e-04, -2.50701269e-05, ...,\n","                   -5.84198511e-04, -5.84395239e-05, -1.84964505e-04]], dtype=float32),\n","           array([-2.3800974e-04,  3.0783727e-04, -7.3823314e-05, ...,\n","                  -8.6034197e-05,  1.2773059e-04,  8.8146036e-05], dtype=float32)),)),\n","        (),\n","        (array([[-4.4390158e-04, -3.0804420e-04, -1.2136692e-03, ...,\n","                 -1.7410841e-04, -8.2126283e-04,  1.0459404e-03],\n","                [ 7.0344679e-05, -8.7649649e-05, -3.6027754e-04, ...,\n","                 -3.4508258e-04,  3.9936334e-05,  3.0654674e-06],\n","                [-1.8798748e-04, -1.6501537e-05, -5.0957658e-04, ...,\n","                 -3.5885336e-05,  3.4958890e-04, -3.6177828e-04],\n","                ...,\n","                [-1.7187247e-04,  3.8567628e-04,  8.3617505e-04, ...,\n","                 -5.6351587e-04, -3.1764567e-04,  7.2552828e-04],\n","                [ 3.9399794e-04,  2.0796966e-04, -4.0387057e-04, ...,\n","                 -2.8239857e-04, -1.8223822e-05,  4.6190900e-05],\n","                [-1.0644007e-03, -4.7872399e-04, -3.1703198e-04, ...,\n","                 -6.0437858e-04,  9.7782619e-04, -2.8006002e-04]], dtype=float32),\n","         array([-2.3647475e-04, -2.5103387e-04,  7.2845163e-05, ...,\n","                 3.3888762e-04, -3.7134072e-05,  1.5325629e-04], dtype=float32))), ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([60.9346  , 42.86501 , 57.105568, ..., 47.913948, 43.974796,\n","              52.084743], dtype=float32),\n","       array([-0.01299958,  0.0055766 ,  0.00381369, ...,  0.00316438,\n","               0.00678433,  0.01253612], dtype=float32)),\n","      (array([[-2.2851274e-04,  3.3882642e-04,  5.8499974e-04, ...,\n","               -3.5416274e-04,  2.5932712e-04,  3.1321144e-04],\n","              [-5.9503895e-05, -2.1542716e-04, -2.8622022e-04, ...,\n","                3.6816249e-04, -6.2046380e-04,  2.0474890e-04],\n","              [-3.5463434e-04, -7.7718485e-04, -8.4459735e-04, ...,\n","               -8.6849468e-05,  2.0617801e-04, -6.1794691e-04],\n","              ...,\n","              [-2.2240197e-04, -7.8765128e-04, -1.4187010e-04, ...,\n","               -1.0148972e-04,  4.4810105e-04,  5.4865965e-04],\n","              [ 1.4189964e-04,  4.5210496e-04, -8.7122782e-04, ...,\n","               -4.2385233e-04, -2.7204392e-04,  3.6592950e-04],\n","              [-1.1616546e-04, -2.4600880e-04, -4.5624169e-04, ...,\n","                1.3337868e-04, -2.0270159e-04, -3.6869862e-04]], dtype=float32),\n","       array([-0.00026152, -0.00031909, -0.00024439, ..., -0.00025571,\n","              -0.00026344, -0.00020376], dtype=float32)),\n","      ((), (), ()),\n","      (),\n","      (array([[-3.9834157e-04,  1.7713461e-04, -8.6412200e-04, ...,\n","                3.1164737e-04,  2.9138103e-04,  6.2306435e-04],\n","              [ 2.8951993e-04, -2.7301064e-04,  9.1393129e-05, ...,\n","                3.3737513e-05,  8.3995104e-04, -4.2753850e-04],\n","              [-2.2627645e-04, -3.2730223e-04, -2.1420832e-05, ...,\n","                6.1842252e-04, -3.5173370e-04, -5.1787315e-04],\n","              ...,\n","              [-4.5683686e-04,  6.2476168e-04, -1.5534610e-04, ...,\n","                2.5628482e-05,  3.5999424e-04, -3.1261612e-04],\n","              [ 2.1145049e-04,  4.2331632e-04,  4.3490515e-04, ...,\n","                4.6223751e-04,  1.0586136e-03, -1.2531897e-04],\n","              [ 1.0547097e-03,  3.3848159e-04,  7.3698134e-04, ...,\n","               -3.0079109e-04,  8.7743002e-04, -6.6376862e-04]], dtype=float32),\n","       array([-2.3812587e-05,  1.3143181e-04,  1.9982814e-04, ...,\n","               1.2848506e-04, -2.7352024e-04, -2.9087544e-04], dtype=float32)),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([16.051727, 14.399047, 15.842695, ..., 13.607024, 17.343388,\n","              15.931021], dtype=float32),\n","       array([ 9.1832690e-04,  2.6306164e-04, -1.1653114e-05, ...,\n","               1.0736884e-04, -2.1401870e-04, -5.0921756e-04], dtype=float32)),\n","      ((),\n","       ((),\n","        (((), (((array([[-1.4303769e-04,  3.2272546e-05, -1.8429260e-04, ...,\n","                     -5.0224608e-04,  2.0016718e-04, -9.8786422e-04],\n","                    [-1.4718683e-04,  8.3212997e-04,  4.3238312e-04, ...,\n","                      1.3489502e-04, -2.4261200e-04, -5.9589365e-04],\n","                    [-1.9882857e-04,  4.5816664e-04, -9.0816716e-04, ...,\n","                     -1.2273930e-03, -1.3664353e-04, -4.8718133e-04],\n","                    ...,\n","                    [-4.9376703e-04,  8.0343668e-04, -1.0963505e-04, ...,\n","                      1.1162120e-03, -3.8362911e-04, -1.6586842e-04],\n","                    [ 1.6119507e-03, -4.7144736e-04,  5.8956136e-04, ...,\n","                     -1.7934333e-03, -5.3616136e-04,  7.5673241e-05],\n","                    [ 2.1724129e-04, -1.3411281e-03,  2.2366174e-04, ...,\n","                     -4.6322300e-04,  3.6323961e-04, -6.7828287e-04]], dtype=float32),\n","             array([ 4.8592068e-05,  1.2348748e-04,  2.3781458e-05, ...,\n","                    -2.4915530e-04, -4.1236691e-04,  1.4609203e-04], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[ 1.3005408e-04,  1.1319956e-03, -1.4622561e-03, ...,\n","                      4.7866508e-04,  3.1612001e-04,  5.8988907e-04],\n","                    [ 2.2839917e-05,  8.5099979e-04,  7.6367037e-04, ...,\n","                     -5.3433666e-04, -3.4718832e-04, -9.2611247e-04],\n","                    [-2.9271643e-04, -1.5609246e-04, -5.7938404e-04, ...,\n","                     -1.4704093e-05,  5.0378230e-04,  4.5935856e-05],\n","                    ...,\n","                    [-3.3143882e-04,  7.4737181e-04, -1.1222671e-03, ...,\n","                      9.9619478e-04, -8.9450664e-04, -7.7410549e-04],\n","                    [ 1.0229113e-03, -7.8080007e-04,  5.4513855e-04, ...,\n","                     -1.5499213e-03,  4.9471582e-04,  8.5206091e-04],\n","                    [-1.0946373e-03, -8.6638425e-04,  6.9588615e-04, ...,\n","                     -1.0247860e-03,  1.6172673e-03,  1.4535137e-03]], dtype=float32),\n","             array([-1.8677108e-04,  3.2834933e-04,  1.1176374e-04, ...,\n","                     3.1993080e-05,  2.9670424e-04, -2.6303731e-04], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[-5.2902298e-05,  5.5748376e-04, -5.9840665e-04, ...,\n","                     -1.6050749e-04,  2.6056811e-04, -3.5743130e-04],\n","                    [-4.6716104e-04,  7.2613836e-04, -2.8673507e-04, ...,\n","                      6.6200300e-05,  5.9534982e-04,  3.3276202e-04],\n","                    [-1.5799329e-03, -5.4600707e-04, -7.9782261e-04, ...,\n","                      1.4005398e-04, -1.8476551e-04, -2.9941593e-04],\n","                    ...,\n","                    [-9.8789786e-04, -3.5856132e-04, -1.9386392e-04, ...,\n","                      3.2746862e-05,  5.1130093e-05,  5.4700905e-04],\n","                    [-2.4352092e-04, -1.5076007e-04, -2.9949215e-04, ...,\n","                     -6.3558604e-05, -2.7712196e-04,  1.7640476e-05],\n","                    [-6.0420961e-04,  5.0304213e-04, -5.2201451e-04, ...,\n","                     -4.5556165e-04,  4.7460521e-04,  2.0233938e-04]], dtype=float32),\n","             array([-5.6764769e-05,  2.3508583e-04,  7.2400901e-05, ...,\n","                     1.7129269e-05, -3.5737943e-05, -1.0323636e-04], dtype=float32)),\n","            ((), (), ())))),\n","         (),\n","         ((), (), ()),\n","         (array([[ 4.05944418e-04,  5.81639179e-04, -1.12171743e-04, ...,\n","                   6.06022193e-04,  1.55891594e-03, -5.18107845e-05],\n","                 [-1.51236032e-04, -2.41307323e-04,  2.06102181e-04, ...,\n","                  -4.50807274e-04, -4.76823014e-04, -2.11694074e-04],\n","                 [-5.13801679e-05, -4.64640209e-04, -5.38857537e-04, ...,\n","                   9.26054432e-04,  7.72669504e-04,  7.97044253e-04],\n","                 ...,\n","                 [ 3.99577548e-04,  1.17500844e-04,  3.22368665e-04, ...,\n","                  -1.68117942e-04, -3.02780099e-04,  8.09859994e-05],\n","                 [-1.20012744e-04,  1.86448553e-04, -1.66430676e-04, ...,\n","                   6.31157382e-05, -2.99310224e-04, -3.95033392e-04],\n","                 [-2.47585180e-04, -4.42452845e-04, -6.21947620e-05, ...,\n","                   2.82516412e-04,  1.74626912e-04,  2.69872893e-04]], dtype=float32),\n","          array([-2.4871330e-04, -2.6331071e-04,  1.2135329e-05, ...,\n","                  2.9343052e-04, -8.3941239e-05,  1.0773823e-04], dtype=float32))),\n","        ()),\n","       ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([24.373844, 23.300625, 24.937374, ..., 22.068832, 18.48495 ,\n","              22.345842], dtype=float32),\n","       array([-4.6662732e-05, -3.0737091e-04, -1.9063464e-04, ...,\n","              -6.1804923e-04,  1.2022637e-04,  1.9711410e-04], dtype=float32)),\n","      (),\n","      ((), (((array([[ 4.0363855e-04, -3.3435429e-04,  4.5642341e-04, ...,\n","                   2.7988816e-04, -2.7804088e-05, -8.0544676e-04],\n","                 [-6.7271903e-04,  6.8655383e-05, -2.1713482e-04, ...,\n","                   4.7166646e-04, -8.4838306e-05,  5.7609833e-04],\n","                 [ 5.0099025e-04, -2.6017238e-04,  3.8617561e-04, ...,\n","                  -1.8507415e-04,  1.4110602e-04, -1.6977701e-04],\n","                 ...,\n","                 [ 6.1249215e-04, -6.0986442e-04, -7.1831088e-04, ...,\n","                   7.0808682e-04,  2.5968158e-04, -9.4781964e-05],\n","                 [-4.6463632e-05, -6.8469747e-04, -4.2867023e-04, ...,\n","                  -1.0882762e-03,  3.8192762e-04, -8.1547746e-04],\n","                 [ 4.7487137e-04, -4.2993386e-04,  7.2464580e-04, ...,\n","                   7.0198243e-05,  6.4003532e-04, -5.1373075e-04]], dtype=float32),\n","          array([-3.8933082e-04,  2.1094175e-04, -1.6936475e-04, ...,\n","                 -1.9018687e-04, -1.1660148e-05,  3.1515679e-05], dtype=float32)),\n","         ((array([[ 1.5895587e-04,  4.3941464e-04, -4.1331758e-04, ...,\n","                   -3.0708231e-04, -6.1755296e-04, -1.3854259e-04],\n","                  [ 1.7138020e-03,  6.6571689e-04,  1.4912316e-03, ...,\n","                    2.7113096e-04,  9.2914305e-04,  8.8725169e-04],\n","                  [ 8.8103943e-06,  6.2321115e-04,  2.8133296e-04, ...,\n","                    3.8010691e-04, -1.0654168e-05, -3.1689560e-05],\n","                  ...,\n","                  [-2.6432207e-04,  1.3816752e-04,  3.2690336e-04, ...,\n","                    2.9140685e-04, -8.5006229e-04,  1.2960939e-03],\n","                  [ 2.5540331e-04,  1.7117575e-04, -1.4458572e-03, ...,\n","                   -6.2124449e-04, -5.6971214e-04, -8.0057379e-04],\n","                  [ 4.1643687e-04,  5.4122595e-04,  8.5178437e-04, ...,\n","                   -8.0393156e-04, -1.1373313e-04, -7.4306357e-04]], dtype=float32),\n","           array([ 8.2542494e-05,  2.3141492e-06,  7.9121608e-05, ...,\n","                  -2.3547477e-04, -3.6451721e-04,  4.6041794e-04], dtype=float32)),),\n","         ((array([[-4.1519103e-04,  6.4417324e-04, -1.5919967e-04, ...,\n","                    9.4118662e-04, -2.1283176e-06, -3.3912004e-04],\n","                  [-6.0132280e-04, -1.5772966e-04, -8.4518273e-05, ...,\n","                   -2.5969133e-04,  1.5388769e-03,  9.8801591e-04],\n","                  [-3.4383102e-04, -1.6974862e-04, -4.0507319e-04, ...,\n","                    3.3051620e-04,  3.1762022e-05, -4.7196914e-05],\n","                  ...,\n","                  [-1.0836212e-03,  4.8211799e-04,  9.6532155e-04, ...,\n","                   -9.2715048e-04,  1.8774248e-03,  1.3621314e-03],\n","                  [-6.3151267e-04,  3.7009668e-05,  2.7561758e-04, ...,\n","                   -1.6896810e-03, -7.9693249e-04,  5.1919313e-04],\n","                  [ 3.2610775e-04, -7.3131209e-04, -2.1062868e-04, ...,\n","                    5.4483917e-05, -2.0671573e-04,  5.8299897e-04]], dtype=float32),\n","           array([-3.2860506e-05, -1.2952604e-04, -4.2809679e-05, ...,\n","                  -1.6456546e-04,  3.7596299e-05, -1.9179151e-04], dtype=float32)),)),\n","        (),\n","        (array([[-2.0203355e-04,  7.9328085e-05, -3.5517663e-04, ...,\n","                  1.5854056e-04, -3.3800240e-04, -2.4718500e-04],\n","                [-3.9318088e-04, -7.0545240e-04,  6.7682756e-04, ...,\n","                  4.0807837e-04, -1.3371331e-04, -3.7258898e-04],\n","                [-1.9211367e-04,  2.7088562e-04, -2.9427500e-04, ...,\n","                 -4.4190211e-04, -4.2002447e-04,  4.1452190e-04],\n","                ...,\n","                [-7.8507781e-04, -1.6217415e-05,  2.7572640e-04, ...,\n","                  2.6987316e-04, -6.3971232e-04,  6.4518076e-04],\n","                [ 3.3510791e-04,  5.3607317e-04, -7.7008888e-05, ...,\n","                  4.6979592e-04, -5.5652478e-04,  4.2632324e-04],\n","                [-8.2728080e-04, -7.6243916e-05,  4.2889413e-04, ...,\n","                 -5.1116181e-04,  1.5377082e-04, -6.5636868e-04]], dtype=float32),\n","         array([-2.01668299e-04, -1.97783767e-04,  1.20525394e-04, ...,\n","                 2.43882998e-04,  8.02269424e-05,  2.45089148e-04], dtype=float32))), ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([52.90552 , 52.82826 , 59.80539 , ..., 59.14005 , 54.50954 ,\n","              53.741776], dtype=float32),\n","       array([-0.00471653, -0.00275178,  0.00248109, ...,  0.00384074,\n","               0.0023349 ,  0.00652673], dtype=float32)),\n","      (array([[-8.1968741e-05,  4.0206421e-04,  5.9266877e-04, ...,\n","               -5.2900042e-04,  4.4751388e-04,  1.1999590e-03],\n","              [-8.3936704e-04, -8.2420134e-05, -1.9481542e-05, ...,\n","                4.1309706e-04, -6.5094006e-04,  3.5036041e-04],\n","              [ 2.4334167e-04, -1.3114880e-03, -2.4131787e-04, ...,\n","               -9.3986360e-05, -6.0818020e-06,  4.0440395e-04],\n","              ...,\n","              [ 5.2142750e-05, -8.2644599e-04, -4.4410693e-04, ...,\n","               -6.4540608e-04, -3.5603700e-04,  4.9932220e-04],\n","              [-4.0718736e-04, -3.4169984e-04, -8.3220453e-05, ...,\n","                1.3175750e-05,  6.8403322e-05, -4.2288541e-04],\n","              [ 4.2111904e-04, -1.8830046e-04,  3.8969902e-05, ...,\n","               -1.5581377e-04, -8.0382306e-04, -1.3438779e-05]], dtype=float32),\n","       array([-0.00023727, -0.00043161, -0.00024709, ..., -0.00021027,\n","              -0.00022718, -0.00021058], dtype=float32)),\n","      ((), (), ()),\n","      (),\n","      (array([[ 2.7933437e-04,  9.2109485e-06, -1.3043813e-04, ...,\n","               -8.5177954e-04,  5.1984440e-05,  5.6357717e-04],\n","              [-1.1815196e-04,  3.3312594e-04, -1.4292037e-04, ...,\n","               -5.1470002e-04, -2.0750586e-04, -4.5727193e-04],\n","              [-1.3451550e-04,  1.0733818e-04, -1.9487320e-04, ...,\n","               -2.2479545e-04, -3.7825856e-04,  8.4281521e-05],\n","              ...,\n","              [-6.7687896e-04,  2.7252494e-05, -2.5610090e-04, ...,\n","               -6.5970497e-05, -9.1204100e-05,  3.7729775e-04],\n","              [ 3.8969834e-04, -4.7901453e-04, -4.6534470e-04, ...,\n","               -1.4069708e-04, -3.7244830e-04,  3.9578139e-05],\n","              [-6.8091118e-04, -3.7613969e-05, -5.7594654e-05, ...,\n","                5.1605428e-04,  4.7481674e-04, -4.2743527e-04]], dtype=float32),\n","       array([ 2.5956591e-05,  2.9341615e-04,  1.1291865e-04, ...,\n","               1.6545673e-04, -3.5097939e-04, -3.5543853e-04], dtype=float32)),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([22.285835, 22.875347, 22.744581, ..., 17.374807, 22.490665,\n","              20.812761], dtype=float32),\n","       array([ 2.8391869e-04,  9.5847936e-05,  1.5093108e-04, ...,\n","              -3.5773762e-04, -8.1025285e-04, -3.3792111e-04], dtype=float32)),\n","      ((),\n","       ((),\n","        (((),\n","          (((array([[ 6.26345340e-04,  7.91497470e-04,  7.05472776e-04, ...,\n","                      1.09953922e-03, -1.18991244e-04, -3.10675777e-03],\n","                    [ 5.29828190e-04,  3.84020968e-04, -7.40046962e-04, ...,\n","                     -8.02746974e-04, -5.01279952e-04,  8.77192651e-04],\n","                    [-2.76243454e-03, -2.63331953e-04,  1.37518073e-04, ...,\n","                     -4.84982098e-04, -5.54896949e-04, -3.22693631e-05],\n","                    ...,\n","                    [-2.36284686e-04, -7.27406645e-04,  8.70721182e-04, ...,\n","                     -4.62369899e-05, -7.86877703e-04,  4.51684347e-04],\n","                    [-1.12042413e-03,  4.81337978e-04,  1.43694109e-03, ...,\n","                      1.30509282e-03,  6.53274474e-04,  9.87833831e-04],\n","                    [-1.89641287e-04,  4.32621891e-04,  1.04647334e-04, ...,\n","                      8.51332297e-05,  2.98614614e-04, -6.74588591e-05]], dtype=float32),\n","             array([ 7.6244716e-05, -2.0459485e-05,  1.1449512e-04, ...,\n","                    -2.6968984e-05,  1.0518397e-05, -8.4850308e-06], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[ 1.2175970e-03,  5.4887071e-04,  6.7672704e-04, ...,\n","                      9.5366169e-04,  8.8173620e-05, -2.3116560e-03],\n","                    [ 4.3011727e-04, -1.2815415e-03, -9.7293936e-04, ...,\n","                     -3.5327752e-04,  3.8569138e-05,  3.5503512e-04],\n","                    [-2.5464850e-03,  1.2184155e-04,  1.0522746e-03, ...,\n","                      3.4934975e-04, -7.6137733e-04, -5.5693078e-04],\n","                    ...,\n","                    [ 4.7763510e-04, -8.9789164e-04,  5.4388941e-04, ...,\n","                      7.9481589e-04, -6.2212866e-04, -4.9768051e-04],\n","                    [-1.4404044e-03,  9.9045690e-04,  2.8418782e-04, ...,\n","                     -2.7179156e-04,  9.2786917e-04,  5.4692739e-04],\n","                    [-1.7126584e-04,  1.1286126e-03, -1.4781163e-04, ...,\n","                      7.9134066e-04, -8.0827775e-04,  1.2004483e-03]], dtype=float32),\n","             array([-1.1525105e-06,  1.7020211e-04,  4.2150219e-04, ...,\n","                    -8.8021792e-05, -6.6898979e-04, -2.6909923e-04], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[-2.52058544e-05,  1.01873658e-04, -1.82281598e-04, ...,\n","                      2.13802006e-04, -3.57469544e-04, -2.73769139e-04],\n","                    [ 1.67720180e-04, -6.80143858e-05,  2.69439333e-04, ...,\n","                     -5.09455334e-04, -1.62739641e-04, -5.38391934e-04],\n","                    [-1.48793464e-04, -9.49793903e-05, -2.48969154e-04, ...,\n","                     -1.09283930e-04, -2.68987205e-04, -5.64054120e-04],\n","                    ...,\n","                    [ 2.81305285e-04,  3.69692862e-04, -5.03625604e-04, ...,\n","                     -4.63599368e-04, -7.46622041e-04, -4.94236592e-04],\n","                    [ 1.48852123e-04,  9.31029208e-05, -7.76082510e-04, ...,\n","                      7.02072983e-04,  1.06214173e-03, -1.74157103e-04],\n","                    [-9.63649421e-04, -3.11392287e-05,  4.08033695e-04, ...,\n","                      1.15905845e-04, -2.54454295e-04, -8.39316926e-05]], dtype=float32),\n","             array([-8.7820482e-04, -2.1079771e-04, -7.3140982e-04, ...,\n","                    -1.6996159e-05, -7.6951314e-05, -9.6892963e-06], dtype=float32)),\n","            ((), (), ())))),\n","         (),\n","         ((), (), ()),\n","         (array([[ 2.8342428e-04, -2.0543457e-04,  1.0154082e-03, ...,\n","                  -1.1881323e-04,  2.2170838e-04,  4.3398535e-04],\n","                 [-2.9596480e-04, -2.5936827e-04,  2.6233515e-04, ...,\n","                   2.8465610e-04, -3.2481988e-04, -1.9243408e-04],\n","                 [ 9.3846073e-05, -2.4958167e-04,  4.3633798e-04, ...,\n","                   4.9830531e-04,  7.4572081e-04,  2.0175186e-04],\n","                 ...,\n","                 [ 2.6756967e-04, -2.3130598e-04,  5.7759351e-04, ...,\n","                   1.2275833e-04, -4.6851780e-04, -1.6825768e-04],\n","                 [-8.7872293e-05, -1.8533236e-04,  2.7986587e-04, ...,\n","                  -4.2824518e-05,  5.0329854e-04,  8.3938095e-04],\n","                 [-1.8140228e-06,  6.1355700e-04, -5.2747567e-04, ...,\n","                   2.0781666e-04,  1.7076856e-04,  1.6101528e-04]], dtype=float32),\n","          array([-3.0376794e-04, -2.8917004e-04,  1.3911238e-04, ...,\n","                  4.2364525e-04,  2.2297337e-04, -7.9336227e-05], dtype=float32))),\n","        ()),\n","       ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([16.017538, 14.684849, 18.17378 , ..., 13.447298, 14.634337,\n","              12.642205], dtype=float32),\n","       array([-2.9349048e-04,  2.0109054e-04, -2.9508365e-04, ...,\n","              -5.9007847e-04,  9.1867420e-05,  1.9200955e-04], dtype=float32)),\n","      (),\n","      ((), (((array([[ 8.2824234e-04,  7.5764758e-05, -3.1308577e-04, ...,\n","                  -7.6474826e-05, -1.0827846e-03, -1.1317811e-03],\n","                 [-3.5891408e-04, -2.3789706e-05,  4.1318362e-04, ...,\n","                  -6.3398114e-04, -3.5971523e-05, -1.0967779e-03],\n","                 [ 5.9075252e-04,  1.3919021e-03, -1.7807563e-04, ...,\n","                  -3.5817878e-04,  1.1353033e-03, -8.3608007e-05],\n","                 ...,\n","                 [-4.7276021e-04,  4.4765449e-04, -1.2320769e-03, ...,\n","                   3.2213415e-04, -1.5893418e-03, -2.1178790e-04],\n","                 [ 6.6952787e-05, -2.3280941e-04, -2.2726832e-04, ...,\n","                   2.5172599e-04,  3.1491110e-04, -2.3875869e-04],\n","                 [-1.2953293e-03, -2.9125702e-04, -6.9298875e-04, ...,\n","                   1.1852060e-03,  4.8431524e-04, -5.3947099e-04]], dtype=float32),\n","          array([-7.7698118e-05,  1.2229042e-04,  2.5959063e-04, ...,\n","                  6.6467631e-04, -7.7643846e-05,  1.8279794e-04], dtype=float32)),\n","         ((array([[-8.0876797e-04,  2.9274091e-04,  4.2400879e-04, ...,\n","                   -3.8956557e-04,  3.7956325e-04, -5.3652271e-04],\n","                  [-1.4144767e-04, -4.7448997e-05, -9.0429862e-04, ...,\n","                    1.0145361e-03, -4.1693958e-04,  2.8129914e-04],\n","                  [ 1.3147500e-04, -2.9552402e-04,  3.6383272e-04, ...,\n","                   -1.1780187e-04, -5.9665024e-04, -2.2316507e-04],\n","                  ...,\n","                  [ 3.8950029e-04,  1.2321402e-03,  4.3418593e-04, ...,\n","                    6.3384720e-04, -1.1671877e-05,  1.8196434e-04],\n","                  [ 9.5754489e-04,  9.3036034e-04, -5.7964696e-04, ...,\n","                    1.2916856e-04,  1.1645566e-03, -6.5747328e-04],\n","                  [ 5.8306090e-04, -7.4161700e-04,  1.7762213e-04, ...,\n","                    8.3760265e-04, -5.8316888e-04,  9.7116770e-04]], dtype=float32),\n","           array([ 1.4688884e-04, -3.7601119e-04,  2.4996358e-05, ...,\n","                   1.3406482e-04,  1.4319230e-04,  4.7508295e-04], dtype=float32)),),\n","         ((array([[-3.9739889e-04,  8.6744738e-05, -3.7913569e-04, ...,\n","                   -2.3590680e-04,  4.4155974e-04,  7.1762479e-04],\n","                  [-4.7125322e-05, -1.0473664e-03, -3.6931789e-04, ...,\n","                    7.1993127e-04, -3.6239749e-04, -1.2021604e-03],\n","                  [-2.7845430e-04, -1.6219738e-04, -5.3785538e-04, ...,\n","                   -2.0259563e-04, -3.0320052e-05, -2.3503981e-04],\n","                  ...,\n","                  [-2.1163901e-04,  1.4155580e-03,  4.6614712e-04, ...,\n","                   -3.0183195e-04, -5.0277973e-04,  9.4633862e-05],\n","                  [ 2.5683932e-05,  6.5858505e-04, -1.1877240e-04, ...,\n","                   -2.8935185e-04,  1.1770087e-03,  8.7497843e-05],\n","                  [-1.6474981e-04,  4.6974432e-04, -5.0981360e-04, ...,\n","                   -1.3395022e-04, -1.4588003e-04,  2.2073693e-04]], dtype=float32),\n","           array([ 3.1843931e-05,  4.2486950e-05,  2.9997253e-05, ...,\n","                  -9.4089577e-05,  1.7028901e-04,  1.9673255e-04], dtype=float32)),)),\n","        (),\n","        (array([[-1.3296629e-04,  3.0773983e-04,  1.2412593e-04, ...,\n","                 -2.4195346e-04, -3.9482937e-04, -4.1650215e-04],\n","                [-7.6760398e-04, -2.4802808e-05, -2.4786952e-04, ...,\n","                 -1.2812122e-04,  9.8122982e-04,  4.5763451e-04],\n","                [-2.7519351e-04,  1.4743314e-04,  1.0389525e-04, ...,\n","                 -6.4129411e-04, -2.6188386e-04,  3.2721946e-05],\n","                ...,\n","                [-3.8106370e-04,  7.3347986e-04, -2.9832372e-04, ...,\n","                 -1.6587970e-04, -3.4595269e-04, -7.3702505e-04],\n","                [ 1.1668538e-04,  3.7801158e-04,  4.2723390e-04, ...,\n","                  3.1585383e-04,  9.4779207e-06,  1.0301641e-03],\n","                [ 4.9032108e-04, -1.1834132e-03,  4.2378274e-04, ...,\n","                  7.3259365e-04,  5.8207894e-04, -1.0510792e-05]], dtype=float32),\n","         array([-2.3185606e-04, -1.2240672e-04,  1.7585584e-04, ...,\n","                 4.4149230e-04,  2.2378023e-04,  2.9431807e-05], dtype=float32))), ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([68.79248 , 54.277626, 60.157932, ..., 64.24441 , 59.326984,\n","              59.93587 ], dtype=float32),\n","       array([-0.00488852, -0.00092812,  0.00130445, ...,  0.00359071,\n","               0.00203295,  0.0055156 ], dtype=float32)),\n","      (array([[ 2.1393900e-04,  2.1183214e-04, -4.4218168e-04, ...,\n","                4.3924374e-05, -3.2522941e-05,  4.5126840e-04],\n","              [ 2.5240477e-04, -4.6099909e-04,  5.1449839e-04, ...,\n","               -1.3662019e-04,  3.3542769e-06, -1.6839030e-04],\n","              [ 3.3757038e-04, -6.8999172e-05, -7.2996411e-04, ...,\n","               -1.7355126e-04, -9.4938866e-04,  2.8393028e-04],\n","              ...,\n","              [ 2.7608674e-04,  3.5430805e-04, -9.0867237e-05, ...,\n","                5.9931423e-04,  9.1966765e-05, -4.4218428e-04],\n","              [-5.2534544e-04, -1.4833611e-04, -3.2956433e-04, ...,\n","               -5.2697235e-04,  2.7104434e-05, -6.1447691e-04],\n","              [-7.1453258e-05, -3.9097166e-04, -8.2851172e-04, ...,\n","                7.4223045e-04,  5.6394911e-04, -3.3672524e-04]], dtype=float32),\n","       array([-0.00013154, -0.00021619, -0.00024289, ..., -0.00024506,\n","              -0.00021154, -0.00018026], dtype=float32)),\n","      ((), (), ()),\n","      (),\n","      (array([[-1.5560248e-04, -6.4064277e-04,  3.8601396e-05, ...,\n","               -4.1018217e-04,  1.2772955e-04,  2.8790833e-04],\n","              [-4.0345863e-04,  7.3224021e-04, -1.3691065e-04, ...,\n","               -4.1101873e-04, -1.7014885e-05, -4.8790203e-04],\n","              [ 5.8300153e-04,  1.7199662e-05,  2.5215783e-04, ...,\n","               -1.1922912e-04,  4.4793313e-04,  6.4021174e-04],\n","              ...,\n","              [-5.8863318e-04,  3.9279155e-04, -5.5593380e-04, ...,\n","               -6.6156063e-04, -9.8348144e-05,  1.8492813e-04],\n","              [-8.0867467e-05,  7.8865545e-05, -6.3464174e-04, ...,\n","                4.8704541e-04, -4.9907097e-04, -9.4332598e-04],\n","              [-8.6980792e-05,  4.1078939e-04,  3.1944248e-04, ...,\n","                1.8304205e-04, -2.3562388e-04,  3.8782312e-04]], dtype=float32),\n","       array([ 1.1229394e-04,  2.8607339e-04, -3.5459871e-05, ...,\n","               2.8196764e-05, -2.7042086e-04, -5.1849493e-04], dtype=float32)),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([17.747932, 17.234734, 16.582468, ..., 16.820122, 17.61388 ,\n","              17.39802 ], dtype=float32),\n","       array([ 2.6117528e-05,  5.7913689e-04, -1.0753079e-04, ...,\n","              -6.5946818e-04, -6.3920807e-04, -4.5076763e-04], dtype=float32)),\n","      ((),\n","       ((),\n","        (((),\n","          (((array([[ 1.56102842e-03,  3.11141630e-04,  1.39255615e-04, ...,\n","                      5.07530698e-04,  3.41892359e-04, -4.67211212e-04],\n","                    [-3.49477545e-04, -6.03729161e-04,  4.06463660e-04, ...,\n","                     -3.94898467e-04, -1.05822710e-05,  9.60061210e-04],\n","                    [ 4.79482085e-04,  3.44425003e-04, -4.83038515e-04, ...,\n","                     -6.59627549e-04, -3.09162308e-04, -5.91234711e-04],\n","                    ...,\n","                    [-8.80545645e-04, -1.16507545e-04, -5.71685378e-04, ...,\n","                      9.44033498e-04,  3.36140278e-04, -2.25783224e-04],\n","                    [-1.23787252e-03, -8.01113376e-04, -1.96791789e-03, ...,\n","                     -1.67309321e-04, -4.69957013e-04, -5.99379491e-05],\n","                    [ 5.10197598e-04,  2.02398351e-03,  1.38677156e-03, ...,\n","                      6.47453184e-04, -7.14988331e-04,  1.25241204e-04]], dtype=float32),\n","             array([2.1426522e-05, 2.4294289e-04, 9.9132296e-05, ..., 1.5201382e-04,\n","                    3.4305299e-04, 2.1296622e-04], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[ 1.20390416e-03,  1.21614919e-03, -1.20033755e-03, ...,\n","                      3.83747247e-04,  6.79096906e-04,  1.10531598e-03],\n","                    [-1.99030444e-04, -8.72096512e-04,  6.84175815e-04, ...,\n","                     -3.39821127e-04,  6.52967545e-04,  2.38339926e-04],\n","                    [ 2.63085170e-03,  1.01917125e-04, -9.05338791e-04, ...,\n","                     -1.31866836e-03,  3.06472823e-04,  9.70645560e-05],\n","                    ...,\n","                    [ 1.42537829e-05, -8.10903963e-04, -7.35808164e-04, ...,\n","                      6.96406118e-04, -3.67396686e-04,  2.35206142e-04],\n","                    [-1.47316349e-03, -1.81162299e-03, -1.49160321e-03, ...,\n","                      1.79246199e-04, -1.07861124e-03,  3.81543388e-04],\n","                    [ 3.96390387e-04,  1.79677596e-03,  1.33570354e-03, ...,\n","                      5.46294206e-04, -8.38863896e-04, -1.56639755e-04]], dtype=float32),\n","             array([-5.1273175e-05, -2.5690574e-04, -1.5539823e-05, ...,\n","                     5.0187518e-04,  4.7923691e-04, -5.2435524e-05], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[-4.9487932e-04,  1.8471619e-04,  1.6571699e-04, ...,\n","                     -5.1121152e-04, -9.9820795e-04, -3.8579214e-04],\n","                    [-2.7290083e-04, -7.3399791e-04, -8.0315204e-04, ...,\n","                     -6.9909974e-04, -1.2292158e-04,  2.3815704e-04],\n","                    [ 4.0994768e-04, -3.3948416e-04,  9.0472220e-04, ...,\n","                     -6.5522239e-04,  2.8979825e-04,  4.0293122e-05],\n","                    ...,\n","                    [ 1.8474071e-04, -3.0772318e-04,  6.5274572e-04, ...,\n","                     -1.5934289e-04, -2.5179129e-04, -3.8195663e-04],\n","                    [-1.5723603e-04, -2.0022484e-04, -6.1404885e-04, ...,\n","                      9.6672651e-04, -1.5702812e-04, -2.9964492e-04],\n","                    [ 3.8635565e-04, -1.0198983e-04,  3.2267351e-05, ...,\n","                      5.9065985e-04,  5.3459971e-04,  4.8879284e-04]], dtype=float32),\n","             array([-6.2742198e-05,  7.1973800e-06,  9.2179696e-05, ...,\n","                    -3.3026125e-05, -1.6200636e-05,  1.2476511e-05], dtype=float32)),\n","            ((), (), ())))),\n","         (),\n","         ((), (), ()),\n","         (array([[ 1.7449810e-04, -2.0221612e-04,  1.2671703e-04, ...,\n","                   2.9711617e-05, -3.3399632e-04,  6.8772829e-04],\n","                 [-2.8090877e-04, -1.0250952e-03, -5.4089871e-04, ...,\n","                  -8.4460923e-04, -2.2784395e-04,  1.3043590e-04],\n","                 [-7.3430863e-05,  3.3030886e-04, -1.5416204e-04, ...,\n","                   6.6254043e-04, -7.5056939e-04, -2.5029547e-04],\n","                 ...,\n","                 [-1.7507022e-04, -1.2997093e-04,  5.7348522e-04, ...,\n","                  -2.7163766e-04,  4.9639103e-04,  7.0256618e-04],\n","                 [-2.4145453e-04, -2.6817189e-04,  4.5067107e-04, ...,\n","                   5.3650740e-04, -5.1721255e-04,  1.0694760e-03],\n","                 [ 1.0201767e-03, -4.8533903e-04,  5.7332421e-05, ...,\n","                   3.0420080e-04, -2.8172327e-04,  1.6563016e-04]], dtype=float32),\n","          array([-1.4108427e-04, -7.3838703e-05,  2.3532208e-04, ...,\n","                  2.3578282e-04,  9.1477639e-05, -1.7532326e-05], dtype=float32))),\n","        ()),\n","       ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([24.317297, 23.844154, 25.127394, ..., 21.20266 , 23.160975,\n","              20.547989], dtype=float32),\n","       array([ 1.3853639e-05,  1.0723363e-04, -1.3101887e-04, ...,\n","              -5.9226010e-04,  1.8432045e-04, -4.3861612e-04], dtype=float32)),\n","      (),\n","      ((), (((array([[-2.8832781e-04,  3.5573461e-04, -2.2991946e-04, ...,\n","                  -1.0914149e-03,  5.2603055e-04,  2.7887931e-04],\n","                 [-9.4513851e-04,  1.3280244e-04, -8.6873135e-04, ...,\n","                  -6.0214405e-04,  9.0189948e-05, -1.9237489e-04],\n","                 [-9.2826493e-05, -2.6180889e-04,  1.5654021e-04, ...,\n","                   5.3430983e-04, -2.6301248e-04, -6.1227602e-04],\n","                 ...,\n","                 [-1.3679339e-03, -1.0553583e-03,  2.2621560e-04, ...,\n","                   3.0303438e-04,  1.2516372e-03,  7.1366772e-04],\n","                 [ 1.3929653e-03, -6.7588344e-04,  3.3959778e-04, ...,\n","                   4.5510373e-05,  3.2150012e-04,  6.4846769e-05],\n","                 [-4.1074317e-04,  2.2440468e-04, -1.3186753e-04, ...,\n","                  -2.0533997e-04, -4.3611444e-04,  7.4847671e-04]], dtype=float32),\n","          array([ 1.1070599e-04, -3.5975102e-04, -1.9818615e-05, ...,\n","                 -2.6786813e-04,  2.0701469e-04, -1.3098678e-04], dtype=float32)),\n","         ((array([[-2.6528924e-05, -4.4999557e-04,  6.0268835e-04, ...,\n","                   -1.1713228e-03,  3.4007753e-05,  9.2294789e-04],\n","                  [-1.4068929e-03,  2.4200877e-05, -5.2252400e-04, ...,\n","                    1.3854954e-03,  4.5505798e-04, -7.4289704e-04],\n","                  [ 4.6430610e-04,  6.8479637e-04,  2.8787259e-04, ...,\n","                   -9.3335722e-05, -4.1503721e-04, -6.0849614e-04],\n","                  ...,\n","                  [ 1.8794027e-04,  7.0611626e-04,  5.4673594e-04, ...,\n","                    7.0684555e-04, -2.8485130e-04, -7.3220406e-04],\n","                  [ 2.5593815e-04, -3.3157851e-04, -7.6794025e-04, ...,\n","                   -4.2171258e-05,  4.3926251e-04,  1.7279853e-03],\n","                  [-7.1588182e-04, -2.7699496e-03,  3.2105122e-04, ...,\n","                    7.6403393e-04,  5.5786414e-04, -8.0063113e-04]], dtype=float32),\n","           array([-3.3926548e-04,  1.4099150e-04, -7.8342651e-05, ...,\n","                  -2.2658589e-05,  2.5869953e-04, -3.9436016e-04], dtype=float32)),),\n","         ((array([[ 5.40972678e-06,  1.25447541e-05, -8.83069879e-05, ...,\n","                   -5.07662480e-04,  9.05934357e-05, -1.07883476e-04],\n","                  [ 5.17993001e-04,  4.44823090e-04,  6.46715751e-04, ...,\n","                    8.63716064e-04,  6.52907824e-04,  7.96378008e-05],\n","                  [ 2.65536451e-04, -3.24084544e-06, -2.21463561e-04, ...,\n","                   -1.35061680e-04,  2.52852187e-04, -3.75624048e-04],\n","                  ...,\n","                  [-7.55664834e-04,  1.51581713e-04,  2.57238426e-04, ...,\n","                    3.93584021e-04,  5.31626807e-04, -5.75449376e-04],\n","                  [ 1.19987031e-04, -1.92989988e-04,  9.70940819e-05, ...,\n","                    4.36141505e-04,  4.00984107e-04,  5.33335377e-04],\n","                  [ 2.45592582e-05,  2.02876210e-04,  3.34093900e-04, ...,\n","                   -2.35376487e-04, -6.07452996e-04, -5.28447505e-04]], dtype=float32),\n","           array([-1.7762643e-04,  8.5051994e-05,  8.1012113e-05, ...,\n","                   1.8490022e-05,  3.1876571e-05,  2.5952872e-05], dtype=float32)),)),\n","        (),\n","        (array([[ 4.11983725e-04,  3.91797657e-04,  6.79950244e-06, ...,\n","                 -2.49070086e-04,  2.96477723e-04,  1.37852214e-03],\n","                [ 4.65474935e-04, -3.66827036e-04,  6.50791219e-04, ...,\n","                  3.03481847e-05,  5.00892813e-04,  4.36378206e-04],\n","                [-1.10610898e-04,  8.28884717e-04,  2.42741255e-04, ...,\n","                 -4.04703227e-04, -1.86379417e-04,  1.94423374e-05],\n","                ...,\n","                [-6.21581159e-04,  2.70798453e-04, -8.63728055e-04, ...,\n","                  1.87227313e-04,  2.18987145e-04, -3.08231422e-04],\n","                [ 1.48002896e-03, -4.33629000e-04, -3.67717410e-04, ...,\n","                  8.08686527e-05,  1.34018046e-04,  4.11196437e-04],\n","                [-3.41945386e-04,  3.26959474e-04, -2.42341193e-04, ...,\n","                  1.13650016e-03, -4.79043665e-04,  1.13544214e-04]], dtype=float32),\n","         array([ 2.7569279e-04, -1.7227685e-04,  1.9234211e-04, ...,\n","                 2.6714004e-04,  1.2580001e-04,  6.7288005e-05], dtype=float32))), ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([63.524796, 47.19608 , 55.53893 , ..., 51.418655, 47.718063,\n","              48.725113], dtype=float32),\n","       array([-7.91660568e-04, -7.46967180e-06,  1.13520604e-04, ...,\n","               8.41949135e-04,  3.77429446e-04,  1.04744011e-03], dtype=float32)),\n","      (array([[-2.6763955e-04, -1.0435457e-04, -3.2954320e-04, ...,\n","                3.6139900e-04, -1.4041651e-04,  6.0938479e-04],\n","              [ 3.8172575e-04, -4.9443496e-04,  9.5118005e-05, ...,\n","                3.7732400e-04,  2.3765054e-04, -5.7034049e-04],\n","              [-5.7377142e-04,  1.3316552e-03, -6.2398141e-04, ...,\n","               -1.9728456e-04,  8.1150509e-05,  5.4578198e-04],\n","              ...,\n","              [-6.1687815e-04,  6.4475636e-05,  4.4190575e-04, ...,\n","               -7.2251831e-04, -6.4342446e-04, -7.7781893e-05],\n","              [-7.8520173e-04, -1.0984247e-03,  4.3555003e-04, ...,\n","               -7.2321591e-06, -9.1663249e-05, -1.2719601e-04],\n","              [-3.4761225e-04,  2.0346553e-04, -4.7609702e-04, ...,\n","               -2.7625175e-04,  2.4562344e-04, -3.5899814e-04]], dtype=float32),\n","       array([-0.00048473, -0.00013151, -0.00025926, ..., -0.00015252,\n","              -0.00013286, -0.00014021], dtype=float32)),\n","      ((), (), ()),\n","      (),\n","      (array([[-3.7430821e-04, -5.6690164e-04, -1.8061402e-04, ...,\n","                2.0781736e-04, -6.9180189e-04, -5.1545782e-04],\n","              [ 1.3990187e-04,  3.9664609e-04,  3.2377269e-04, ...,\n","               -6.3897553e-04,  3.4341984e-04,  1.8947310e-04],\n","              [ 4.0012950e-04,  4.7675708e-06, -2.2281353e-04, ...,\n","               -3.6916297e-04, -4.3977884e-04,  1.2107067e-05],\n","              ...,\n","              [-1.5048186e-05, -3.5379222e-04, -3.0002752e-04, ...,\n","                4.2823493e-04,  1.7223405e-04,  5.5495551e-04],\n","              [-7.2019681e-04, -6.4449816e-04, -4.3714949e-04, ...,\n","               -4.1086704e-04, -1.4792364e-05,  1.6051989e-04],\n","              [ 5.2630249e-04,  3.7332618e-04,  2.2256462e-04, ...,\n","               -3.6538549e-05,  2.0231833e-04, -1.2506853e-04]], dtype=float32),\n","       array([ 0.00045243,  0.00041044, -0.00032689, ..., -0.00018825,\n","              -0.00043491, -0.00058477], dtype=float32)),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([13.752513, 15.128455, 16.433582, ..., 16.912975, 13.715821,\n","              15.958053], dtype=float32),\n","       array([ 0.00083225,  0.00091359, -0.00063647, ...,  0.00027756,\n","              -0.00075828, -0.00087873], dtype=float32)),\n","      ((),\n","       ((),\n","        (((), (((array([[ 9.5709151e-04, -5.2387902e-04, -3.1958215e-04, ...,\n","                     -6.2054634e-04,  6.8096246e-04,  4.1744357e-04],\n","                    [-7.2381590e-05,  8.0718273e-05,  1.1026340e-03, ...,\n","                     -3.1608777e-04,  4.5228604e-04, -4.2267068e-04],\n","                    [ 1.6897825e-03, -6.7114597e-04, -1.3457355e-03, ...,\n","                     -3.9492513e-04,  6.5481634e-04, -1.3097553e-03],\n","                    ...,\n","                    [ 1.9926708e-03, -1.1131020e-03, -1.1497344e-04, ...,\n","                      8.8179263e-04, -9.2443440e-04, -1.1420862e-03],\n","                    [-1.9340987e-03,  2.3192244e-03,  6.0904905e-04, ...,\n","                     -1.0641922e-04,  6.9369169e-05, -8.9647265e-06],\n","                    [ 8.1459369e-04, -1.0087213e-03, -4.7072142e-05, ...,\n","                      5.5514864e-04, -3.2097823e-04,  1.0499733e-03]], dtype=float32),\n","             array([-3.88440618e-04, -9.34294003e-05,  1.47283063e-04, ...,\n","                     6.43572013e-04, -7.10880384e-04, -1.04402214e-04], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[ 1.5499708e-03, -4.9026124e-04, -2.9873839e-04, ...,\n","                     -6.9657981e-06, -6.6335994e-04,  1.0234591e-03],\n","                    [ 5.2048662e-04, -6.1513920e-04,  8.3564088e-04, ...,\n","                      8.3501288e-04, -6.1360822e-04,  1.4358906e-04],\n","                    [ 1.4965929e-03,  8.8006008e-04, -2.8151274e-04, ...,\n","                     -4.8701843e-04,  1.6847282e-04,  2.9762791e-04],\n","                    ...,\n","                    [ 2.0989464e-03,  2.2231300e-04, -3.5890882e-04, ...,\n","                     -3.2680022e-05, -3.5992239e-04, -1.5249162e-03],\n","                    [-5.4875934e-05,  7.9933292e-04,  3.8840223e-04, ...,\n","                     -8.5689861e-04,  1.6636420e-03,  5.6419050e-04],\n","                    [ 2.2247715e-03, -2.6484890e-04, -5.6789722e-04, ...,\n","                     -5.7143415e-04,  6.1653584e-04, -3.7242641e-04]], dtype=float32),\n","             array([ 4.3184384e-05, -8.7316068e-05, -1.9974042e-04, ...,\n","                     1.1264901e-04, -2.4886153e-04,  1.7487718e-04], dtype=float32)),\n","            ((), (), ())),\n","           ((array([[ 8.4612322e-05, -3.9380297e-04, -4.9169490e-04, ...,\n","                     -3.4935522e-04, -7.3786854e-04, -5.1246985e-04],\n","                    [ 4.9152126e-04,  3.6586987e-04,  4.5507011e-04, ...,\n","                      1.3109130e-03,  8.4423262e-04, -5.3623575e-04],\n","                    [ 2.0919141e-04,  2.5720394e-04, -4.3592072e-04, ...,\n","                     -1.6832010e-04, -1.0311357e-05,  1.3579967e-04],\n","                    ...,\n","                    [ 4.8325624e-04,  2.7567911e-04,  1.3809542e-04, ...,\n","                      1.9216766e-05,  7.0527720e-04,  7.2328985e-04],\n","                    [-4.7356519e-04, -4.9043167e-04, -1.9397885e-04, ...,\n","                     -2.4736332e-04,  3.5232020e-05,  6.0783850e-04],\n","                    [ 1.2550669e-04,  3.2946817e-04, -1.9481471e-04, ...,\n","                     -1.3942758e-03, -5.3641613e-04, -8.0123224e-04]], dtype=float32),\n","             array([ 1.21520636e-04, -1.63897188e-04,  2.34490013e-04, ...,\n","                     7.72461208e-05,  3.11622571e-05,  4.14756360e-04], dtype=float32)),\n","            ((), (), ())))),\n","         (),\n","         ((), (), ()),\n","         (array([[ 1.0785917e-04,  2.6424613e-04,  8.3731960e-05, ...,\n","                   3.2483842e-04, -5.6673482e-04,  1.3477192e-04],\n","                 [-4.9799182e-05, -3.2693605e-04, -1.9544664e-04, ...,\n","                  -8.3103491e-04,  3.8955148e-04, -6.5902638e-04],\n","                 [-5.1327376e-04, -2.5788868e-05, -3.9683437e-04, ...,\n","                   2.4251956e-04,  7.0035756e-05, -4.0077767e-04],\n","                 ...,\n","                 [ 3.0524028e-04,  7.2996959e-04, -6.8823877e-04, ...,\n","                   2.0093573e-04,  3.0612358e-05,  6.6998566e-04],\n","                 [-1.1261798e-03,  1.5622820e-04, -1.2443489e-03, ...,\n","                  -2.1155383e-03, -1.2917203e-04, -7.7054260e-04],\n","                 [ 3.6069134e-05, -8.6956308e-04,  2.8569277e-04, ...,\n","                  -2.5763584e-04, -7.1995593e-05,  1.0159860e-04]], dtype=float32),\n","          array([ 2.3879729e-05, -1.7653120e-04,  1.5005236e-04, ...,\n","                 -4.7831621e-05,  1.8492920e-04, -3.5481571e-04], dtype=float32))),\n","        ()),\n","       ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([13.798606, 13.067237, 11.609435, ..., 13.882713, 10.965513,\n","               8.080246], dtype=float32),\n","       array([-6.01322972e-05,  1.38413627e-04, -1.35315553e-04, ...,\n","              -5.82156645e-04,  1.02778555e-04,  1.03747298e-04], dtype=float32)),\n","      (),\n","      ((), (((array([[-3.9936139e-04, -1.1795694e-03,  4.9896876e-04, ...,\n","                   1.3181023e-04,  1.4290026e-04,  2.7917488e-04],\n","                 [ 7.0126663e-04, -6.6100206e-04,  3.9345384e-04, ...,\n","                   1.2796404e-04,  3.3058631e-04, -1.8799644e-04],\n","                 [ 1.0917565e-03,  5.5716647e-04,  4.7140627e-04, ...,\n","                  -2.8474766e-04, -4.3198772e-04,  9.2382508e-04],\n","                 ...,\n","                 [ 7.2382340e-05, -1.3853029e-04,  3.0214980e-04, ...,\n","                  -4.4249094e-04,  5.1563291e-04,  6.9918111e-04],\n","                 [ 7.7126111e-04,  8.6039113e-04,  2.2489100e-04, ...,\n","                   1.1876241e-03, -3.5500611e-04, -2.7336195e-04],\n","                 [-9.7407820e-04,  5.8198383e-04,  5.7858870e-05, ...,\n","                   8.7042473e-04, -1.0395362e-03, -1.1996158e-04]], dtype=float32),\n","          array([ 1.2035250e-04,  2.6425047e-04,  1.6716718e-04, ...,\n","                 -1.5004881e-04,  2.2141190e-05,  3.0131897e-04], dtype=float32)),\n","         ((array([[-6.2232011e-04,  8.2496245e-04, -2.3138533e-04, ...,\n","                   -6.2539504e-04,  9.3390699e-05,  4.1927458e-04],\n","                  [-2.0739937e-04,  7.3603075e-04,  1.0017530e-03, ...,\n","                    2.7482348e-04,  8.5724541e-04,  1.0473697e-03],\n","                  [-4.6838223e-04, -2.2335614e-04, -4.3492252e-04, ...,\n","                   -3.8435595e-05,  9.7381481e-04,  2.3159223e-04],\n","                  ...,\n","                  [ 2.8937394e-04, -1.2386631e-03,  2.8099932e-04, ...,\n","                   -1.2159722e-05,  7.1644451e-04, -2.4572155e-04],\n","                  [ 9.0166915e-04,  5.7993957e-04,  6.7580730e-04, ...,\n","                    9.8813209e-04,  4.0605047e-04, -6.5798237e-04],\n","                  [ 7.9643534e-04,  9.9252420e-04,  1.5070307e-05, ...,\n","                   -4.8876478e-05,  3.1102466e-04, -4.6027664e-04]], dtype=float32),\n","           array([-1.4925725e-04,  7.3104212e-07,  2.4234348e-04, ...,\n","                   1.8831350e-04,  4.9156439e-05, -3.1956536e-04], dtype=float32)),),\n","         ((array([[-5.1481952e-04,  1.8094247e-04,  8.3509643e-05, ...,\n","                   -4.0844854e-05, -1.1113316e-03,  5.5797398e-04],\n","                  [-2.9111141e-04, -1.7374296e-05, -5.9384468e-04, ...,\n","                   -8.1645278e-04,  3.2181598e-04, -3.7493272e-04],\n","                  [ 1.5219238e-05, -2.4851123e-04, -4.0863489e-04, ...,\n","                    2.2453876e-04, -1.5173161e-05,  4.6613536e-04],\n","                  ...,\n","                  [ 6.7562144e-04,  4.1817311e-05,  1.8628474e-04, ...,\n","                   -9.6429302e-04,  1.1025828e-03, -5.5980642e-04],\n","                  [ 7.7820080e-04,  3.4929722e-04, -6.4103195e-04, ...,\n","                   -2.1449431e-04, -9.9557021e-04,  5.9047324e-04],\n","                  [ 4.0733835e-04,  1.8405083e-04,  3.2022268e-05, ...,\n","                    2.7981555e-04, -8.4180880e-04, -1.9624905e-04]], dtype=float32),\n","           array([-0.0002401 , -0.00017576, -0.0001405 , ..., -0.00012909,\n","                   0.00047414,  0.00033517], dtype=float32)),)),\n","        (),\n","        (array([[ 1.0998470e-04, -4.0206989e-05,  4.1721622e-04, ...,\n","                 -2.8197991e-04, -3.8833948e-04,  2.3209080e-04],\n","                [ 6.5347290e-04,  7.6249370e-04, -5.9591199e-04, ...,\n","                  3.7964515e-04, -1.3393204e-03,  2.2020617e-04],\n","                [ 8.9442980e-04,  1.2371418e-04,  1.1953693e-05, ...,\n","                 -4.8702516e-04,  2.5688394e-04,  3.5516558e-05],\n","                ...,\n","                [-5.3574820e-04, -2.6228116e-04, -5.8096839e-04, ...,\n","                 -5.4698763e-04, -4.2329755e-04, -3.8424056e-04],\n","                [ 7.3962954e-05,  2.1569799e-04,  2.2385734e-04, ...,\n","                 -2.0282491e-04,  1.3272432e-04,  8.2439679e-04],\n","                [-2.1870702e-04,  5.0568971e-04, -8.7354559e-04, ...,\n","                  1.3916404e-04, -7.5441314e-04, -1.1015392e-04]], dtype=float32),\n","         array([ 9.5154835e-05, -1.2709547e-04,  3.3921853e-05, ...,\n","                 1.3296401e-04, -7.1882969e-05,  7.0729264e-05], dtype=float32))), ()),\n","      ()))),\n","   ()),\n","  (((),\n","    ((), ((array([27.491665, 24.65252 , 21.665281, ..., 23.682236, 22.432665,\n","              25.401508], dtype=float32),\n","       array([ 0.00050309,  0.00042072, -0.00096551, ...,  0.00014515,\n","              -0.00024783,  0.00017171], dtype=float32)),\n","      (array([[ 6.1336026e-04,  3.1313823e-05,  9.8945224e-04, ...,\n","               -4.0913365e-04,  3.2328258e-04, -6.6883536e-04],\n","              [-1.4023170e-04,  4.5154951e-04,  4.9215590e-04, ...,\n","                1.4820666e-03,  8.2416204e-04, -1.6435495e-04],\n","              [-4.8850728e-05, -2.9825454e-04, -3.3648332e-04, ...,\n","               -1.0867490e-04, -1.4428642e-04, -3.2859965e-04],\n","              ...,\n","              [-1.6962238e-04, -1.4840379e-04, -1.7255399e-04, ...,\n","                3.8138652e-04, -6.6862631e-05, -6.3153583e-04],\n","              [ 5.1564019e-04,  1.8929597e-04,  2.4282196e-04, ...,\n","                3.5733671e-04,  2.0580135e-05, -2.4422747e-04],\n","              [-3.1826735e-04,  1.2306521e-03, -2.4065550e-04, ...,\n","                1.4719566e-04,  6.9295208e-04, -4.6351350e-05]], dtype=float32),\n","       array([ 1.5015337e-04, -3.2263095e-04,  7.9013778e-05, ...,\n","              -2.7027397e-04, -5.5851058e-05,  3.2064994e-04], dtype=float32)),\n","      ((), (), ()),\n","      (),\n","      (array([[-6.6430453e-04, -2.6509253e-04,  4.3522925e-04, ...,\n","               -5.4013199e-06,  3.0094641e-04, -4.3674110e-05],\n","              [ 8.2220597e-04,  2.4146432e-04,  2.9450608e-04, ...,\n","               -9.1872993e-04, -7.4948580e-04, -7.8093557e-04],\n","              [ 4.4695835e-04,  2.4359330e-04, -2.8813328e-04, ...,\n","               -3.1769523e-04, -4.4001321e-05, -2.0626881e-04],\n","              ...,\n","              [ 1.3570056e-03, -1.2622436e-04,  6.8024109e-04, ...,\n","               -3.4084712e-04, -7.5046759e-04,  2.0607692e-04],\n","              [ 9.6459058e-04, -1.8928487e-04,  2.0946315e-04, ...,\n","               -8.6039194e-04,  1.6756915e-04,  5.3349423e-04],\n","              [-1.3755567e-04,  3.7209815e-04,  1.0722101e-03, ...,\n","               -5.0377572e-04,  8.2195766e-05,  1.4802278e-04]], dtype=float32),\n","       array([ 0.00073418,  0.00050296, -0.00057693, ..., -0.00028076,\n","              -0.00063786, -0.00042135], dtype=float32)),\n","      ()))),\n","   ()),\n","  (array([124.08609, 145.1185 , 131.66975, ..., 143.21477, 145.33452,\n","          139.15051], dtype=float32),\n","   array([-5.4492844e-05, -3.5730217e-04,  1.0062612e-03, ...,\n","           3.2377170e-04,  8.1990851e-04,  4.4442255e-05], dtype=float32)),\n","  (),\n","  (array([[ 5.1487889e-04, -1.8631468e-04, -3.4254947e-04, ...,\n","           -4.1661315e-04, -1.3360759e-04,  2.2999300e-03],\n","          [ 5.2118841e-05,  3.5506525e-04,  1.9759311e-04, ...,\n","            7.1740570e-04, -1.6157045e-04, -5.1092234e-04],\n","          [-6.0540368e-04,  5.0488469e-04,  3.6138089e-04, ...,\n","           -4.7829797e-04, -7.3525502e-04, -1.0206304e-03],\n","          ...,\n","          [-2.0523426e-04,  2.8229301e-04, -1.8258688e-04, ...,\n","           -1.6772746e-04, -5.2244228e-04, -1.3218011e-04],\n","          [-5.6004158e-04, -6.9031979e-05,  2.4686265e-04, ...,\n","           -2.7989177e-04, -1.9914578e-04, -2.6545988e-04],\n","          [-4.4482123e-04, -3.2657653e-04, -1.1193550e-04, ...,\n","           -4.4081049e-04,  1.3587816e-04, -6.1564217e-04]], dtype=float32),\n","   array([-0.00446677,  0.00093652,  0.00062758, ...,  0.00055092,\n","           0.00053187,  0.00095175], dtype=float32))),\n"," ((),\n","  ((), ((), ((), (), ()))),\n","  ((),\n","   ((),\n","    DeviceArray(0.1, dtype=float32),\n","    (),\n","    (((),\n","      ((),\n","       ((),\n","        ((), ((), ((), (((), (), ()), (), ()), ())), ()),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        (),\n","        ((), (), ()),\n","        DeviceArray(0.1, dtype=float32),\n","        (),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        ((), ((), ((), (((), (), ()), (), ()), ())), ()),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        (),\n","        ((), (), ()),\n","        DeviceArray(0.1, dtype=float32),\n","        (),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        ((), ((), ((), (((), (), ()), (), ()), ())), ()),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        (),\n","        ((), (), ()),\n","        DeviceArray(0.1, dtype=float32),\n","        (),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        ((), ((), ((), (((), (), ()), (), ()), ())), ()),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        (),\n","        ((), (), ()),\n","        DeviceArray(0.1, dtype=float32),\n","        (),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        ((), ((), ((), (((), (), ()), (), ()), ())), ()),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        (),\n","        ((), (), ()),\n","        DeviceArray(0.1, dtype=float32),\n","        (),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        ((), ((), ((), (((), (), ()), (), ()), ())), ()),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        (),\n","        ((), (), ()),\n","        DeviceArray(0.1, dtype=float32),\n","        (),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        ((), ((), ((), (((), (), ()), (), ()), ())), ()),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        (),\n","        ((), (), ()),\n","        DeviceArray(0.1, dtype=float32),\n","        (),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        ((), ((), ((), (((), (), ()), (), ()), ())), ()),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        (),\n","        ((), (), ()),\n","        DeviceArray(0.1, dtype=float32),\n","        (),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        ((), ((), ((), (((), (), ()), (), ()), ())), ()),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        (),\n","        ((), (), ()),\n","        DeviceArray(0.1, dtype=float32),\n","        (),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        ((), ((), ((), (((), (), ()), (), ()), ())), ()),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        (),\n","        ((), (), ()),\n","        DeviceArray(0.1, dtype=float32),\n","        (),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        ((), ((), ((), (((), (), ()), (), ()), ())), ()),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        (),\n","        ((), (), ()),\n","        DeviceArray(0.1, dtype=float32),\n","        (),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        ((), ((), ((), (((), (), ()), (), ()), ())), ()),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        (),\n","        ((), (), ()),\n","        DeviceArray(0.1, dtype=float32),\n","        (),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        ((), ((), ((), (((), (), ()), (), ()), ())), ()),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        (),\n","        ((), (), ()),\n","        DeviceArray(0.1, dtype=float32),\n","        (),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        ((), ((), ((), (((), (), ()), (), ()), ())), ()),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        (),\n","        ((), (), ()),\n","        DeviceArray(0.1, dtype=float32),\n","        (),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        ((), ((), ((), (((), (), ()), (), ()), ())), ()),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        (),\n","        ((), (), ()),\n","        DeviceArray(0.1, dtype=float32),\n","        (),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        ((), ((), ((), (((), (), ()), (), ()), ())), ()),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        (),\n","        ((), (), ()),\n","        DeviceArray(0.1, dtype=float32),\n","        (),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        ((), ((), ((), (((), (), ()), (), ()), ())), ()),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        (),\n","        ((), (), ()),\n","        DeviceArray(0.1, dtype=float32),\n","        (),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        ((), ((), ((), (((), (), ()), (), ()), ())), ()),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        (),\n","        ((), (), ()),\n","        DeviceArray(0.1, dtype=float32),\n","        (),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        ((), ((), ((), (((), (), ()), (), ()), ())), ()),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        (),\n","        ((), (), ()),\n","        DeviceArray(0.1, dtype=float32),\n","        (),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        ((), ((), ((), (((), (), ()), (), ()), ())), ()),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        (),\n","        ((), (), ()),\n","        DeviceArray(0.1, dtype=float32),\n","        (),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        ((), ((), ((), (((), (), ()), (), ()), ())), ()),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        (),\n","        ((), (), ()),\n","        DeviceArray(0.1, dtype=float32),\n","        (),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        ((), ((), ((), (((), (), ()), (), ()), ())), ()),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        (),\n","        ((), (), ()),\n","        DeviceArray(0.1, dtype=float32),\n","        (),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        ((), ((), ((), (((), (), ()), (), ()), ())), ()),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        (),\n","        ((), (), ()),\n","        DeviceArray(0.1, dtype=float32),\n","        (),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        ((), ((), ((), (((), (), ()), (), ()), ())), ()),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    (((),\n","      ((),\n","       ((),\n","        (),\n","        ((), (), ()),\n","        DeviceArray(0.1, dtype=float32),\n","        (),\n","        DeviceArray(0.1, dtype=float32)))),\n","     ()),\n","    ())),\n","  (),\n","  ((), (), ()),\n","  {'__marker_for_cached_state_': ()},\n","  DeviceArray(0.1, dtype=float32),\n","  DeviceArray(0, dtype=int32),\n","  ((), ((), ())),\n","  (((),\n","    ((),\n","     ((),\n","      ((),\n","       ((),\n","        (((), (((), ((), (), ())), ((), ((), (), ())), ((), ((), (), ())))),\n","         (DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray(0, dtype=int32)),\n","         ((), (), ()),\n","         ()),\n","        ()),\n","       ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (((), ((), ()), ((), ())), (), ()), ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (), ()),\n","      DeviceArray(0.1, dtype=float32),\n","      (),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      ((),\n","       ((),\n","        (((), (((), ((), (), ())), ((), ((), (), ())), ((), ((), (), ())))),\n","         (DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray(0, dtype=int32)),\n","         ((), (), ()),\n","         ()),\n","        ()),\n","       ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (((), ((), ()), ((), ())), (), ()), ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (), ()),\n","      DeviceArray(0.1, dtype=float32),\n","      (),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      ((),\n","       ((),\n","        (((), (((), ((), (), ())), ((), ((), (), ())), ((), ((), (), ())))),\n","         (DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray(0, dtype=int32)),\n","         ((), (), ()),\n","         ()),\n","        ()),\n","       ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (((), ((), ()), ((), ())), (), ()), ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (), ()),\n","      DeviceArray(0.1, dtype=float32),\n","      (),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      ((),\n","       ((),\n","        (((), (((), ((), (), ())), ((), ((), (), ())), ((), ((), (), ())))),\n","         (DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray(0, dtype=int32)),\n","         ((), (), ()),\n","         ()),\n","        ()),\n","       ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (((), ((), ()), ((), ())), (), ()), ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (), ()),\n","      DeviceArray(0.1, dtype=float32),\n","      (),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      ((),\n","       ((),\n","        (((), (((), ((), (), ())), ((), ((), (), ())), ((), ((), (), ())))),\n","         (DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray(0, dtype=int32)),\n","         ((), (), ()),\n","         ()),\n","        ()),\n","       ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (((), ((), ()), ((), ())), (), ()), ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (), ()),\n","      DeviceArray(0.1, dtype=float32),\n","      (),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      ((),\n","       ((),\n","        (((), (((), ((), (), ())), ((), ((), (), ())), ((), ((), (), ())))),\n","         (DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray(0, dtype=int32)),\n","         ((), (), ()),\n","         ()),\n","        ()),\n","       ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (((), ((), ()), ((), ())), (), ()), ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (), ()),\n","      DeviceArray(0.1, dtype=float32),\n","      (),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      ((),\n","       ((),\n","        (((), (((), ((), (), ())), ((), ((), (), ())), ((), ((), (), ())))),\n","         (DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray(0, dtype=int32)),\n","         ((), (), ()),\n","         ()),\n","        ()),\n","       ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (((), ((), ()), ((), ())), (), ()), ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (), ()),\n","      DeviceArray(0.1, dtype=float32),\n","      (),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      ((),\n","       ((),\n","        (((), (((), ((), (), ())), ((), ((), (), ())), ((), ((), (), ())))),\n","         (DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray(0, dtype=int32)),\n","         ((), (), ()),\n","         ()),\n","        ()),\n","       ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (((), ((), ()), ((), ())), (), ()), ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (), ()),\n","      DeviceArray(0.1, dtype=float32),\n","      (),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      ((),\n","       ((),\n","        (((), (((), ((), (), ())), ((), ((), (), ())), ((), ((), (), ())))),\n","         (DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray(0, dtype=int32)),\n","         ((), (), ()),\n","         ()),\n","        ()),\n","       ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (((), ((), ()), ((), ())), (), ()), ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (), ()),\n","      DeviceArray(0.1, dtype=float32),\n","      (),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      ((),\n","       ((),\n","        (((), (((), ((), (), ())), ((), ((), (), ())), ((), ((), (), ())))),\n","         (DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray(0, dtype=int32)),\n","         ((), (), ()),\n","         ()),\n","        ()),\n","       ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (((), ((), ()), ((), ())), (), ()), ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (), ()),\n","      DeviceArray(0.1, dtype=float32),\n","      (),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      ((),\n","       ((),\n","        (((), (((), ((), (), ())), ((), ((), (), ())), ((), ((), (), ())))),\n","         (DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray(0, dtype=int32)),\n","         ((), (), ()),\n","         ()),\n","        ()),\n","       ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (((), ((), ()), ((), ())), (), ()), ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (), ()),\n","      DeviceArray(0.1, dtype=float32),\n","      (),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      ((),\n","       ((),\n","        (((), (((), ((), (), ())), ((), ((), (), ())), ((), ((), (), ())))),\n","         (DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray(0, dtype=int32)),\n","         ((), (), ()),\n","         ()),\n","        ()),\n","       ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (((), ((), ()), ((), ())), (), ()), ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (), ()),\n","      DeviceArray(0.1, dtype=float32),\n","      (),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      ((),\n","       ((),\n","        (((), (((), ((), (), ())), ((), ((), (), ())), ((), ((), (), ())))),\n","         (DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray(0, dtype=int32)),\n","         ((), (), ()),\n","         ()),\n","        ()),\n","       ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (((), ((), ()), ((), ())), (), ()), ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (), ()),\n","      DeviceArray(0.1, dtype=float32),\n","      (),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      ((),\n","       ((),\n","        (((), (((), ((), (), ())), ((), ((), (), ())), ((), ((), (), ())))),\n","         (DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray(0, dtype=int32)),\n","         ((), (), ()),\n","         ()),\n","        ()),\n","       ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (((), ((), ()), ((), ())), (), ()), ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (), ()),\n","      DeviceArray(0.1, dtype=float32),\n","      (),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      ((),\n","       ((),\n","        (((), (((), ((), (), ())), ((), ((), (), ())), ((), ((), (), ())))),\n","         (DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray(0, dtype=int32)),\n","         ((), (), ()),\n","         ()),\n","        ()),\n","       ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (((), ((), ()), ((), ())), (), ()), ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (), ()),\n","      DeviceArray(0.1, dtype=float32),\n","      (),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      ((),\n","       ((),\n","        (((), (((), ((), (), ())), ((), ((), (), ())), ((), ((), (), ())))),\n","         (DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray(0, dtype=int32)),\n","         ((), (), ()),\n","         ()),\n","        ()),\n","       ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (((), ((), ()), ((), ())), (), ()), ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (), ()),\n","      DeviceArray(0.1, dtype=float32),\n","      (),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      ((),\n","       ((),\n","        (((), (((), ((), (), ())), ((), ((), (), ())), ((), ((), (), ())))),\n","         (DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray(0, dtype=int32)),\n","         ((), (), ()),\n","         ()),\n","        ()),\n","       ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (((), ((), ()), ((), ())), (), ()), ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (), ()),\n","      DeviceArray(0.1, dtype=float32),\n","      (),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      ((),\n","       ((),\n","        (((), (((), ((), (), ())), ((), ((), (), ())), ((), ((), (), ())))),\n","         (DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray(0, dtype=int32)),\n","         ((), (), ()),\n","         ()),\n","        ()),\n","       ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (((), ((), ()), ((), ())), (), ()), ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (), ()),\n","      DeviceArray(0.1, dtype=float32),\n","      (),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      ((),\n","       ((),\n","        (((), (((), ((), (), ())), ((), ((), (), ())), ((), ((), (), ())))),\n","         (DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray(0, dtype=int32)),\n","         ((), (), ()),\n","         ()),\n","        ()),\n","       ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (((), ((), ()), ((), ())), (), ()), ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (), ()),\n","      DeviceArray(0.1, dtype=float32),\n","      (),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      ((),\n","       ((),\n","        (((), (((), ((), (), ())), ((), ((), (), ())), ((), ((), (), ())))),\n","         (DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray(0, dtype=int32)),\n","         ((), (), ()),\n","         ()),\n","        ()),\n","       ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (((), ((), ()), ((), ())), (), ()), ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (), ()),\n","      DeviceArray(0.1, dtype=float32),\n","      (),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      ((),\n","       ((),\n","        (((), (((), ((), (), ())), ((), ((), (), ())), ((), ((), (), ())))),\n","         (DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray(0, dtype=int32)),\n","         ((), (), ()),\n","         ()),\n","        ()),\n","       ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (((), ((), ()), ((), ())), (), ()), ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (), ()),\n","      DeviceArray(0.1, dtype=float32),\n","      (),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      ((),\n","       ((),\n","        (((), (((), ((), (), ())), ((), ((), (), ())), ((), ((), (), ())))),\n","         (DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray(0, dtype=int32)),\n","         ((), (), ()),\n","         ()),\n","        ()),\n","       ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (((), ((), ()), ((), ())), (), ()), ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (), ()),\n","      DeviceArray(0.1, dtype=float32),\n","      (),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      ((),\n","       ((),\n","        (((), (((), ((), (), ())), ((), ((), (), ())), ((), ((), (), ())))),\n","         (DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray(0, dtype=int32)),\n","         ((), (), ()),\n","         ()),\n","        ()),\n","       ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (((), ((), ()), ((), ())), (), ()), ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (), ()),\n","      DeviceArray(0.1, dtype=float32),\n","      (),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      ((),\n","       ((),\n","        (((), (((), ((), (), ())), ((), ((), (), ())), ((), ((), (), ())))),\n","         (DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray([[[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       ...,\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]],\n","          \n","                       [[0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        ...,\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.],\n","                        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32),\n","          DeviceArray(0, dtype=int32)),\n","         ((), (), ()),\n","         ()),\n","        ()),\n","       ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (((), ((), ()), ((), ())), (), ()), ()),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (((),\n","    ((),\n","     ((),\n","      (),\n","      ((), (), ()),\n","      DeviceArray(0.1, dtype=float32),\n","      (),\n","      DeviceArray(0.1, dtype=float32)))),\n","   ()),\n","  (),\n","  (),\n","  ()))"]},"metadata":{"tags":[]},"execution_count":10}]},{"cell_type":"code","metadata":{"id":"l33umWrRyYNm","colab":{"base_uri":"https://localhost:8080/"},"executionInfo":{"status":"ok","timestamp":1624414019257,"user_tz":-420,"elapsed":753,"user":{"displayName":"Quý Trần Nhật","photoUrl":"","userId":"13689480600881371104"}},"outputId":"4f516619-5a59-4216-e6ac-1ff870fda222"},"source":["# Uncomment to see the transformer's structure.\n","print(model)"],"execution_count":11,"outputs":[{"output_type":"stream","text":["Serial_in2_out2[\n","  Select[0,1,1]_in2_out3\n","  Branch_out2[\n","    []\n","    Serial[\n","      PaddingMask(0)\n","    ]\n","  ]\n","  Cache_in2_out2[\n","    Serial_in2_out2[\n","      Embedding_32000_1024\n","      Dropout\n","      PositionalEncoding\n","      Serial_in2_out2[\n","        Branch_in2_out3[\n","          None\n","          Serial_in2_out2[\n","            LayerNorm\n","            Serial_in2_out2[\n","              _in2_out2\n","              Serial_in2_out2[\n","                Select[0,0,0]_out3\n","                Serial_in4_out2[\n","                  _in4_out4\n","                  Serial_in4_out2[\n","                    Parallel_in3_out3[\n","                      Dense_1024\n","                      Dense_1024\n","                      Dense_1024\n","                    ]\n","                    PureAttention_in4_out2\n","                    Dense_1024\n","                  ]\n","                  _in2_out2\n","                ]\n","              ]\n","              _in2_out2\n","            ]\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial[\n","        Branch_out2[\n","          None\n","          Serial[\n","            LayerNorm\n","            Dense_4096\n","            Serial[\n","              Relu\n","            ]\n","            Dropout\n","            Dense_1024\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial_in2_out2[\n","        Branch_in2_out3[\n","          None\n","          Serial_in2_out2[\n","            LayerNorm\n","            Serial_in2_out2[\n","              _in2_out2\n","              Serial_in2_out2[\n","                Select[0,0,0]_out3\n","                Serial_in4_out2[\n","                  _in4_out4\n","                  Serial_in4_out2[\n","                    Parallel_in3_out3[\n","                      Dense_1024\n","                      Dense_1024\n","                      Dense_1024\n","                    ]\n","                    PureAttention_in4_out2\n","                    Dense_1024\n","                  ]\n","                  _in2_out2\n","                ]\n","              ]\n","              _in2_out2\n","            ]\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial[\n","        Branch_out2[\n","          None\n","          Serial[\n","            LayerNorm\n","            Dense_4096\n","            Serial[\n","              Relu\n","            ]\n","            Dropout\n","            Dense_1024\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial_in2_out2[\n","        Branch_in2_out3[\n","          None\n","          Serial_in2_out2[\n","            LayerNorm\n","            Serial_in2_out2[\n","              _in2_out2\n","              Serial_in2_out2[\n","                Select[0,0,0]_out3\n","                Serial_in4_out2[\n","                  _in4_out4\n","                  Serial_in4_out2[\n","                    Parallel_in3_out3[\n","                      Dense_1024\n","                      Dense_1024\n","                      Dense_1024\n","                    ]\n","                    PureAttention_in4_out2\n","                    Dense_1024\n","                  ]\n","                  _in2_out2\n","                ]\n","              ]\n","              _in2_out2\n","            ]\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial[\n","        Branch_out2[\n","          None\n","          Serial[\n","            LayerNorm\n","            Dense_4096\n","            Serial[\n","              Relu\n","            ]\n","            Dropout\n","            Dense_1024\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial_in2_out2[\n","        Branch_in2_out3[\n","          None\n","          Serial_in2_out2[\n","            LayerNorm\n","            Serial_in2_out2[\n","              _in2_out2\n","              Serial_in2_out2[\n","                Select[0,0,0]_out3\n","                Serial_in4_out2[\n","                  _in4_out4\n","                  Serial_in4_out2[\n","                    Parallel_in3_out3[\n","                      Dense_1024\n","                      Dense_1024\n","                      Dense_1024\n","                    ]\n","                    PureAttention_in4_out2\n","                    Dense_1024\n","                  ]\n","                  _in2_out2\n","                ]\n","              ]\n","              _in2_out2\n","            ]\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial[\n","        Branch_out2[\n","          None\n","          Serial[\n","            LayerNorm\n","            Dense_4096\n","            Serial[\n","              Relu\n","            ]\n","            Dropout\n","            Dense_1024\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial_in2_out2[\n","        Branch_in2_out3[\n","          None\n","          Serial_in2_out2[\n","            LayerNorm\n","            Serial_in2_out2[\n","              _in2_out2\n","              Serial_in2_out2[\n","                Select[0,0,0]_out3\n","                Serial_in4_out2[\n","                  _in4_out4\n","                  Serial_in4_out2[\n","                    Parallel_in3_out3[\n","                      Dense_1024\n","                      Dense_1024\n","                      Dense_1024\n","                    ]\n","                    PureAttention_in4_out2\n","                    Dense_1024\n","                  ]\n","                  _in2_out2\n","                ]\n","              ]\n","              _in2_out2\n","            ]\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial[\n","        Branch_out2[\n","          None\n","          Serial[\n","            LayerNorm\n","            Dense_4096\n","            Serial[\n","              Relu\n","            ]\n","            Dropout\n","            Dense_1024\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial_in2_out2[\n","        Branch_in2_out3[\n","          None\n","          Serial_in2_out2[\n","            LayerNorm\n","            Serial_in2_out2[\n","              _in2_out2\n","              Serial_in2_out2[\n","                Select[0,0,0]_out3\n","                Serial_in4_out2[\n","                  _in4_out4\n","                  Serial_in4_out2[\n","                    Parallel_in3_out3[\n","                      Dense_1024\n","                      Dense_1024\n","                      Dense_1024\n","                    ]\n","                    PureAttention_in4_out2\n","                    Dense_1024\n","                  ]\n","                  _in2_out2\n","                ]\n","              ]\n","              _in2_out2\n","            ]\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial[\n","        Branch_out2[\n","          None\n","          Serial[\n","            LayerNorm\n","            Dense_4096\n","            Serial[\n","              Relu\n","            ]\n","            Dropout\n","            Dense_1024\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial_in2_out2[\n","        Branch_in2_out3[\n","          None\n","          Serial_in2_out2[\n","            LayerNorm\n","            Serial_in2_out2[\n","              _in2_out2\n","              Serial_in2_out2[\n","                Select[0,0,0]_out3\n","                Serial_in4_out2[\n","                  _in4_out4\n","                  Serial_in4_out2[\n","                    Parallel_in3_out3[\n","                      Dense_1024\n","                      Dense_1024\n","                      Dense_1024\n","                    ]\n","                    PureAttention_in4_out2\n","                    Dense_1024\n","                  ]\n","                  _in2_out2\n","                ]\n","              ]\n","              _in2_out2\n","            ]\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial[\n","        Branch_out2[\n","          None\n","          Serial[\n","            LayerNorm\n","            Dense_4096\n","            Serial[\n","              Relu\n","            ]\n","            Dropout\n","            Dense_1024\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial_in2_out2[\n","        Branch_in2_out3[\n","          None\n","          Serial_in2_out2[\n","            LayerNorm\n","            Serial_in2_out2[\n","              _in2_out2\n","              Serial_in2_out2[\n","                Select[0,0,0]_out3\n","                Serial_in4_out2[\n","                  _in4_out4\n","                  Serial_in4_out2[\n","                    Parallel_in3_out3[\n","                      Dense_1024\n","                      Dense_1024\n","                      Dense_1024\n","                    ]\n","                    PureAttention_in4_out2\n","                    Dense_1024\n","                  ]\n","                  _in2_out2\n","                ]\n","              ]\n","              _in2_out2\n","            ]\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial[\n","        Branch_out2[\n","          None\n","          Serial[\n","            LayerNorm\n","            Dense_4096\n","            Serial[\n","              Relu\n","            ]\n","            Dropout\n","            Dense_1024\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial_in2_out2[\n","        Branch_in2_out3[\n","          None\n","          Serial_in2_out2[\n","            LayerNorm\n","            Serial_in2_out2[\n","              _in2_out2\n","              Serial_in2_out2[\n","                Select[0,0,0]_out3\n","                Serial_in4_out2[\n","                  _in4_out4\n","                  Serial_in4_out2[\n","                    Parallel_in3_out3[\n","                      Dense_1024\n","                      Dense_1024\n","                      Dense_1024\n","                    ]\n","                    PureAttention_in4_out2\n","                    Dense_1024\n","                  ]\n","                  _in2_out2\n","                ]\n","              ]\n","              _in2_out2\n","            ]\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial[\n","        Branch_out2[\n","          None\n","          Serial[\n","            LayerNorm\n","            Dense_4096\n","            Serial[\n","              Relu\n","            ]\n","            Dropout\n","            Dense_1024\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial_in2_out2[\n","        Branch_in2_out3[\n","          None\n","          Serial_in2_out2[\n","            LayerNorm\n","            Serial_in2_out2[\n","              _in2_out2\n","              Serial_in2_out2[\n","                Select[0,0,0]_out3\n","                Serial_in4_out2[\n","                  _in4_out4\n","                  Serial_in4_out2[\n","                    Parallel_in3_out3[\n","                      Dense_1024\n","                      Dense_1024\n","                      Dense_1024\n","                    ]\n","                    PureAttention_in4_out2\n","                    Dense_1024\n","                  ]\n","                  _in2_out2\n","                ]\n","              ]\n","              _in2_out2\n","            ]\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial[\n","        Branch_out2[\n","          None\n","          Serial[\n","            LayerNorm\n","            Dense_4096\n","            Serial[\n","              Relu\n","            ]\n","            Dropout\n","            Dense_1024\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial_in2_out2[\n","        Branch_in2_out3[\n","          None\n","          Serial_in2_out2[\n","            LayerNorm\n","            Serial_in2_out2[\n","              _in2_out2\n","              Serial_in2_out2[\n","                Select[0,0,0]_out3\n","                Serial_in4_out2[\n","                  _in4_out4\n","                  Serial_in4_out2[\n","                    Parallel_in3_out3[\n","                      Dense_1024\n","                      Dense_1024\n","                      Dense_1024\n","                    ]\n","                    PureAttention_in4_out2\n","                    Dense_1024\n","                  ]\n","                  _in2_out2\n","                ]\n","              ]\n","              _in2_out2\n","            ]\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial[\n","        Branch_out2[\n","          None\n","          Serial[\n","            LayerNorm\n","            Dense_4096\n","            Serial[\n","              Relu\n","            ]\n","            Dropout\n","            Dense_1024\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial_in2_out2[\n","        Branch_in2_out3[\n","          None\n","          Serial_in2_out2[\n","            LayerNorm\n","            Serial_in2_out2[\n","              _in2_out2\n","              Serial_in2_out2[\n","                Select[0,0,0]_out3\n","                Serial_in4_out2[\n","                  _in4_out4\n","                  Serial_in4_out2[\n","                    Parallel_in3_out3[\n","                      Dense_1024\n","                      Dense_1024\n","                      Dense_1024\n","                    ]\n","                    PureAttention_in4_out2\n","                    Dense_1024\n","                  ]\n","                  _in2_out2\n","                ]\n","              ]\n","              _in2_out2\n","            ]\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial[\n","        Branch_out2[\n","          None\n","          Serial[\n","            LayerNorm\n","            Dense_4096\n","            Serial[\n","              Relu\n","            ]\n","            Dropout\n","            Dense_1024\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial_in2_out2[\n","        Branch_in2_out3[\n","          None\n","          Serial_in2_out2[\n","            LayerNorm\n","            Serial_in2_out2[\n","              _in2_out2\n","              Serial_in2_out2[\n","                Select[0,0,0]_out3\n","                Serial_in4_out2[\n","                  _in4_out4\n","                  Serial_in4_out2[\n","                    Parallel_in3_out3[\n","                      Dense_1024\n","                      Dense_1024\n","                      Dense_1024\n","                    ]\n","                    PureAttention_in4_out2\n","                    Dense_1024\n","                  ]\n","                  _in2_out2\n","                ]\n","              ]\n","              _in2_out2\n","            ]\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial[\n","        Branch_out2[\n","          None\n","          Serial[\n","            LayerNorm\n","            Dense_4096\n","            Serial[\n","              Relu\n","            ]\n","            Dropout\n","            Dense_1024\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial_in2_out2[\n","        Branch_in2_out3[\n","          None\n","          Serial_in2_out2[\n","            LayerNorm\n","            Serial_in2_out2[\n","              _in2_out2\n","              Serial_in2_out2[\n","                Select[0,0,0]_out3\n","                Serial_in4_out2[\n","                  _in4_out4\n","                  Serial_in4_out2[\n","                    Parallel_in3_out3[\n","                      Dense_1024\n","                      Dense_1024\n","                      Dense_1024\n","                    ]\n","                    PureAttention_in4_out2\n","                    Dense_1024\n","                  ]\n","                  _in2_out2\n","                ]\n","              ]\n","              _in2_out2\n","            ]\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial[\n","        Branch_out2[\n","          None\n","          Serial[\n","            LayerNorm\n","            Dense_4096\n","            Serial[\n","              Relu\n","            ]\n","            Dropout\n","            Dense_1024\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial_in2_out2[\n","        Branch_in2_out3[\n","          None\n","          Serial_in2_out2[\n","            LayerNorm\n","            Serial_in2_out2[\n","              _in2_out2\n","              Serial_in2_out2[\n","                Select[0,0,0]_out3\n","                Serial_in4_out2[\n","                  _in4_out4\n","                  Serial_in4_out2[\n","                    Parallel_in3_out3[\n","                      Dense_1024\n","                      Dense_1024\n","                      Dense_1024\n","                    ]\n","                    PureAttention_in4_out2\n","                    Dense_1024\n","                  ]\n","                  _in2_out2\n","                ]\n","              ]\n","              _in2_out2\n","            ]\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial[\n","        Branch_out2[\n","          None\n","          Serial[\n","            LayerNorm\n","            Dense_4096\n","            Serial[\n","              Relu\n","            ]\n","            Dropout\n","            Dense_1024\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial_in2_out2[\n","        Branch_in2_out3[\n","          None\n","          Serial_in2_out2[\n","            LayerNorm\n","            Serial_in2_out2[\n","              _in2_out2\n","              Serial_in2_out2[\n","                Select[0,0,0]_out3\n","                Serial_in4_out2[\n","                  _in4_out4\n","                  Serial_in4_out2[\n","                    Parallel_in3_out3[\n","                      Dense_1024\n","                      Dense_1024\n","                      Dense_1024\n","                    ]\n","                    PureAttention_in4_out2\n","                    Dense_1024\n","                  ]\n","                  _in2_out2\n","                ]\n","              ]\n","              _in2_out2\n","            ]\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial[\n","        Branch_out2[\n","          None\n","          Serial[\n","            LayerNorm\n","            Dense_4096\n","            Serial[\n","              Relu\n","            ]\n","            Dropout\n","            Dense_1024\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial_in2_out2[\n","        Branch_in2_out3[\n","          None\n","          Serial_in2_out2[\n","            LayerNorm\n","            Serial_in2_out2[\n","              _in2_out2\n","              Serial_in2_out2[\n","                Select[0,0,0]_out3\n","                Serial_in4_out2[\n","                  _in4_out4\n","                  Serial_in4_out2[\n","                    Parallel_in3_out3[\n","                      Dense_1024\n","                      Dense_1024\n","                      Dense_1024\n","                    ]\n","                    PureAttention_in4_out2\n","                    Dense_1024\n","                  ]\n","                  _in2_out2\n","                ]\n","              ]\n","              _in2_out2\n","            ]\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial[\n","        Branch_out2[\n","          None\n","          Serial[\n","            LayerNorm\n","            Dense_4096\n","            Serial[\n","              Relu\n","            ]\n","            Dropout\n","            Dense_1024\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial_in2_out2[\n","        Branch_in2_out3[\n","          None\n","          Serial_in2_out2[\n","            LayerNorm\n","            Serial_in2_out2[\n","              _in2_out2\n","              Serial_in2_out2[\n","                Select[0,0,0]_out3\n","                Serial_in4_out2[\n","                  _in4_out4\n","                  Serial_in4_out2[\n","                    Parallel_in3_out3[\n","                      Dense_1024\n","                      Dense_1024\n","                      Dense_1024\n","                    ]\n","                    PureAttention_in4_out2\n","                    Dense_1024\n","                  ]\n","                  _in2_out2\n","                ]\n","              ]\n","              _in2_out2\n","            ]\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial[\n","        Branch_out2[\n","          None\n","          Serial[\n","            LayerNorm\n","            Dense_4096\n","            Serial[\n","              Relu\n","            ]\n","            Dropout\n","            Dense_1024\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial_in2_out2[\n","        Branch_in2_out3[\n","          None\n","          Serial_in2_out2[\n","            LayerNorm\n","            Serial_in2_out2[\n","              _in2_out2\n","              Serial_in2_out2[\n","                Select[0,0,0]_out3\n","                Serial_in4_out2[\n","                  _in4_out4\n","                  Serial_in4_out2[\n","                    Parallel_in3_out3[\n","                      Dense_1024\n","                      Dense_1024\n","                      Dense_1024\n","                    ]\n","                    PureAttention_in4_out2\n","                    Dense_1024\n","                  ]\n","                  _in2_out2\n","                ]\n","              ]\n","              _in2_out2\n","            ]\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial[\n","        Branch_out2[\n","          None\n","          Serial[\n","            LayerNorm\n","            Dense_4096\n","            Serial[\n","              Relu\n","            ]\n","            Dropout\n","            Dense_1024\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial_in2_out2[\n","        Branch_in2_out3[\n","          None\n","          Serial_in2_out2[\n","            LayerNorm\n","            Serial_in2_out2[\n","              _in2_out2\n","              Serial_in2_out2[\n","                Select[0,0,0]_out3\n","                Serial_in4_out2[\n","                  _in4_out4\n","                  Serial_in4_out2[\n","                    Parallel_in3_out3[\n","                      Dense_1024\n","                      Dense_1024\n","                      Dense_1024\n","                    ]\n","                    PureAttention_in4_out2\n","                    Dense_1024\n","                  ]\n","                  _in2_out2\n","                ]\n","              ]\n","              _in2_out2\n","            ]\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial[\n","        Branch_out2[\n","          None\n","          Serial[\n","            LayerNorm\n","            Dense_4096\n","            Serial[\n","              Relu\n","            ]\n","            Dropout\n","            Dense_1024\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial_in2_out2[\n","        Branch_in2_out3[\n","          None\n","          Serial_in2_out2[\n","            LayerNorm\n","            Serial_in2_out2[\n","              _in2_out2\n","              Serial_in2_out2[\n","                Select[0,0,0]_out3\n","                Serial_in4_out2[\n","                  _in4_out4\n","                  Serial_in4_out2[\n","                    Parallel_in3_out3[\n","                      Dense_1024\n","                      Dense_1024\n","                      Dense_1024\n","                    ]\n","                    PureAttention_in4_out2\n","                    Dense_1024\n","                  ]\n","                  _in2_out2\n","                ]\n","              ]\n","              _in2_out2\n","            ]\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial[\n","        Branch_out2[\n","          None\n","          Serial[\n","            LayerNorm\n","            Dense_4096\n","            Serial[\n","              Relu\n","            ]\n","            Dropout\n","            Dense_1024\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial_in2_out2[\n","        Branch_in2_out3[\n","          None\n","          Serial_in2_out2[\n","            LayerNorm\n","            Serial_in2_out2[\n","              _in2_out2\n","              Serial_in2_out2[\n","                Select[0,0,0]_out3\n","                Serial_in4_out2[\n","                  _in4_out4\n","                  Serial_in4_out2[\n","                    Parallel_in3_out3[\n","                      Dense_1024\n","                      Dense_1024\n","                      Dense_1024\n","                    ]\n","                    PureAttention_in4_out2\n","                    Dense_1024\n","                  ]\n","                  _in2_out2\n","                ]\n","              ]\n","              _in2_out2\n","            ]\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial[\n","        Branch_out2[\n","          None\n","          Serial[\n","            LayerNorm\n","            Dense_4096\n","            Serial[\n","              Relu\n","            ]\n","            Dropout\n","            Dense_1024\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial_in2_out2[\n","        Branch_in2_out3[\n","          None\n","          Serial_in2_out2[\n","            LayerNorm\n","            Serial_in2_out2[\n","              _in2_out2\n","              Serial_in2_out2[\n","                Select[0,0,0]_out3\n","                Serial_in4_out2[\n","                  _in4_out4\n","                  Serial_in4_out2[\n","                    Parallel_in3_out3[\n","                      Dense_1024\n","                      Dense_1024\n","                      Dense_1024\n","                    ]\n","                    PureAttention_in4_out2\n","                    Dense_1024\n","                  ]\n","                  _in2_out2\n","                ]\n","              ]\n","              _in2_out2\n","            ]\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial[\n","        Branch_out2[\n","          None\n","          Serial[\n","            LayerNorm\n","            Dense_4096\n","            Serial[\n","              Relu\n","            ]\n","            Dropout\n","            Dense_1024\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial_in2_out2[\n","        Branch_in2_out3[\n","          None\n","          Serial_in2_out2[\n","            LayerNorm\n","            Serial_in2_out2[\n","              _in2_out2\n","              Serial_in2_out2[\n","                Select[0,0,0]_out3\n","                Serial_in4_out2[\n","                  _in4_out4\n","                  Serial_in4_out2[\n","                    Parallel_in3_out3[\n","                      Dense_1024\n","                      Dense_1024\n","                      Dense_1024\n","                    ]\n","                    PureAttention_in4_out2\n","                    Dense_1024\n","                  ]\n","                  _in2_out2\n","                ]\n","              ]\n","              _in2_out2\n","            ]\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      Serial[\n","        Branch_out2[\n","          None\n","          Serial[\n","            LayerNorm\n","            Dense_4096\n","            Serial[\n","              Relu\n","            ]\n","            Dropout\n","            Dense_1024\n","            Dropout\n","          ]\n","        ]\n","        Add_in2\n","      ]\n","      LayerNorm\n","    ]\n","  ]\n","  Select[2,1,0]_in3_out3\n","  Serial[\n","    ShiftRight(1)\n","  ]\n","  Embedding_32000_1024\n","  Dropout\n","  PositionalEncoding\n","  Branch_in2_out2[\n","    []\n","    EncoderDecoderMask_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Serial[\n","          Serial[\n","            Serial[\n","              Branch_out3[\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","              ]\n","              DotProductCausalAttention_in3\n","              Serial[\n","                MergeHeads\n","              ]\n","              Dense_1024\n","            ]\n","          ]\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial_in3_out3[\n","    Branch_in3_out4[\n","      None\n","      Serial_in3_out3[\n","        LayerNorm\n","        Select[0,2,2,1,2]_in3_out5\n","        Serial_in4_out2[\n","          _in4_out4\n","          Serial_in4_out2[\n","            Parallel_in3_out3[\n","              Dense_1024\n","              Cache[\n","                Dense_1024\n","              ]\n","              Cache[\n","                Dense_1024\n","              ]\n","            ]\n","            PureAttention_in4_out2\n","            Dense_1024\n","          ]\n","          _in2_out2\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Dense_4096\n","        Serial[\n","          Relu\n","        ]\n","        Dropout\n","        Dense_1024\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Serial[\n","          Serial[\n","            Serial[\n","              Branch_out3[\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","              ]\n","              DotProductCausalAttention_in3\n","              Serial[\n","                MergeHeads\n","              ]\n","              Dense_1024\n","            ]\n","          ]\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial_in3_out3[\n","    Branch_in3_out4[\n","      None\n","      Serial_in3_out3[\n","        LayerNorm\n","        Select[0,2,2,1,2]_in3_out5\n","        Serial_in4_out2[\n","          _in4_out4\n","          Serial_in4_out2[\n","            Parallel_in3_out3[\n","              Dense_1024\n","              Cache[\n","                Dense_1024\n","              ]\n","              Cache[\n","                Dense_1024\n","              ]\n","            ]\n","            PureAttention_in4_out2\n","            Dense_1024\n","          ]\n","          _in2_out2\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Dense_4096\n","        Serial[\n","          Relu\n","        ]\n","        Dropout\n","        Dense_1024\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Serial[\n","          Serial[\n","            Serial[\n","              Branch_out3[\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","              ]\n","              DotProductCausalAttention_in3\n","              Serial[\n","                MergeHeads\n","              ]\n","              Dense_1024\n","            ]\n","          ]\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial_in3_out3[\n","    Branch_in3_out4[\n","      None\n","      Serial_in3_out3[\n","        LayerNorm\n","        Select[0,2,2,1,2]_in3_out5\n","        Serial_in4_out2[\n","          _in4_out4\n","          Serial_in4_out2[\n","            Parallel_in3_out3[\n","              Dense_1024\n","              Cache[\n","                Dense_1024\n","              ]\n","              Cache[\n","                Dense_1024\n","              ]\n","            ]\n","            PureAttention_in4_out2\n","            Dense_1024\n","          ]\n","          _in2_out2\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Dense_4096\n","        Serial[\n","          Relu\n","        ]\n","        Dropout\n","        Dense_1024\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Serial[\n","          Serial[\n","            Serial[\n","              Branch_out3[\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","              ]\n","              DotProductCausalAttention_in3\n","              Serial[\n","                MergeHeads\n","              ]\n","              Dense_1024\n","            ]\n","          ]\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial_in3_out3[\n","    Branch_in3_out4[\n","      None\n","      Serial_in3_out3[\n","        LayerNorm\n","        Select[0,2,2,1,2]_in3_out5\n","        Serial_in4_out2[\n","          _in4_out4\n","          Serial_in4_out2[\n","            Parallel_in3_out3[\n","              Dense_1024\n","              Cache[\n","                Dense_1024\n","              ]\n","              Cache[\n","                Dense_1024\n","              ]\n","            ]\n","            PureAttention_in4_out2\n","            Dense_1024\n","          ]\n","          _in2_out2\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Dense_4096\n","        Serial[\n","          Relu\n","        ]\n","        Dropout\n","        Dense_1024\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Serial[\n","          Serial[\n","            Serial[\n","              Branch_out3[\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","              ]\n","              DotProductCausalAttention_in3\n","              Serial[\n","                MergeHeads\n","              ]\n","              Dense_1024\n","            ]\n","          ]\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial_in3_out3[\n","    Branch_in3_out4[\n","      None\n","      Serial_in3_out3[\n","        LayerNorm\n","        Select[0,2,2,1,2]_in3_out5\n","        Serial_in4_out2[\n","          _in4_out4\n","          Serial_in4_out2[\n","            Parallel_in3_out3[\n","              Dense_1024\n","              Cache[\n","                Dense_1024\n","              ]\n","              Cache[\n","                Dense_1024\n","              ]\n","            ]\n","            PureAttention_in4_out2\n","            Dense_1024\n","          ]\n","          _in2_out2\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Dense_4096\n","        Serial[\n","          Relu\n","        ]\n","        Dropout\n","        Dense_1024\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Serial[\n","          Serial[\n","            Serial[\n","              Branch_out3[\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","              ]\n","              DotProductCausalAttention_in3\n","              Serial[\n","                MergeHeads\n","              ]\n","              Dense_1024\n","            ]\n","          ]\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial_in3_out3[\n","    Branch_in3_out4[\n","      None\n","      Serial_in3_out3[\n","        LayerNorm\n","        Select[0,2,2,1,2]_in3_out5\n","        Serial_in4_out2[\n","          _in4_out4\n","          Serial_in4_out2[\n","            Parallel_in3_out3[\n","              Dense_1024\n","              Cache[\n","                Dense_1024\n","              ]\n","              Cache[\n","                Dense_1024\n","              ]\n","            ]\n","            PureAttention_in4_out2\n","            Dense_1024\n","          ]\n","          _in2_out2\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Dense_4096\n","        Serial[\n","          Relu\n","        ]\n","        Dropout\n","        Dense_1024\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Serial[\n","          Serial[\n","            Serial[\n","              Branch_out3[\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","              ]\n","              DotProductCausalAttention_in3\n","              Serial[\n","                MergeHeads\n","              ]\n","              Dense_1024\n","            ]\n","          ]\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial_in3_out3[\n","    Branch_in3_out4[\n","      None\n","      Serial_in3_out3[\n","        LayerNorm\n","        Select[0,2,2,1,2]_in3_out5\n","        Serial_in4_out2[\n","          _in4_out4\n","          Serial_in4_out2[\n","            Parallel_in3_out3[\n","              Dense_1024\n","              Cache[\n","                Dense_1024\n","              ]\n","              Cache[\n","                Dense_1024\n","              ]\n","            ]\n","            PureAttention_in4_out2\n","            Dense_1024\n","          ]\n","          _in2_out2\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Dense_4096\n","        Serial[\n","          Relu\n","        ]\n","        Dropout\n","        Dense_1024\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Serial[\n","          Serial[\n","            Serial[\n","              Branch_out3[\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","              ]\n","              DotProductCausalAttention_in3\n","              Serial[\n","                MergeHeads\n","              ]\n","              Dense_1024\n","            ]\n","          ]\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial_in3_out3[\n","    Branch_in3_out4[\n","      None\n","      Serial_in3_out3[\n","        LayerNorm\n","        Select[0,2,2,1,2]_in3_out5\n","        Serial_in4_out2[\n","          _in4_out4\n","          Serial_in4_out2[\n","            Parallel_in3_out3[\n","              Dense_1024\n","              Cache[\n","                Dense_1024\n","              ]\n","              Cache[\n","                Dense_1024\n","              ]\n","            ]\n","            PureAttention_in4_out2\n","            Dense_1024\n","          ]\n","          _in2_out2\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Dense_4096\n","        Serial[\n","          Relu\n","        ]\n","        Dropout\n","        Dense_1024\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Serial[\n","          Serial[\n","            Serial[\n","              Branch_out3[\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","              ]\n","              DotProductCausalAttention_in3\n","              Serial[\n","                MergeHeads\n","              ]\n","              Dense_1024\n","            ]\n","          ]\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial_in3_out3[\n","    Branch_in3_out4[\n","      None\n","      Serial_in3_out3[\n","        LayerNorm\n","        Select[0,2,2,1,2]_in3_out5\n","        Serial_in4_out2[\n","          _in4_out4\n","          Serial_in4_out2[\n","            Parallel_in3_out3[\n","              Dense_1024\n","              Cache[\n","                Dense_1024\n","              ]\n","              Cache[\n","                Dense_1024\n","              ]\n","            ]\n","            PureAttention_in4_out2\n","            Dense_1024\n","          ]\n","          _in2_out2\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Dense_4096\n","        Serial[\n","          Relu\n","        ]\n","        Dropout\n","        Dense_1024\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Serial[\n","          Serial[\n","            Serial[\n","              Branch_out3[\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","              ]\n","              DotProductCausalAttention_in3\n","              Serial[\n","                MergeHeads\n","              ]\n","              Dense_1024\n","            ]\n","          ]\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial_in3_out3[\n","    Branch_in3_out4[\n","      None\n","      Serial_in3_out3[\n","        LayerNorm\n","        Select[0,2,2,1,2]_in3_out5\n","        Serial_in4_out2[\n","          _in4_out4\n","          Serial_in4_out2[\n","            Parallel_in3_out3[\n","              Dense_1024\n","              Cache[\n","                Dense_1024\n","              ]\n","              Cache[\n","                Dense_1024\n","              ]\n","            ]\n","            PureAttention_in4_out2\n","            Dense_1024\n","          ]\n","          _in2_out2\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Dense_4096\n","        Serial[\n","          Relu\n","        ]\n","        Dropout\n","        Dense_1024\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Serial[\n","          Serial[\n","            Serial[\n","              Branch_out3[\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","              ]\n","              DotProductCausalAttention_in3\n","              Serial[\n","                MergeHeads\n","              ]\n","              Dense_1024\n","            ]\n","          ]\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial_in3_out3[\n","    Branch_in3_out4[\n","      None\n","      Serial_in3_out3[\n","        LayerNorm\n","        Select[0,2,2,1,2]_in3_out5\n","        Serial_in4_out2[\n","          _in4_out4\n","          Serial_in4_out2[\n","            Parallel_in3_out3[\n","              Dense_1024\n","              Cache[\n","                Dense_1024\n","              ]\n","              Cache[\n","                Dense_1024\n","              ]\n","            ]\n","            PureAttention_in4_out2\n","            Dense_1024\n","          ]\n","          _in2_out2\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Dense_4096\n","        Serial[\n","          Relu\n","        ]\n","        Dropout\n","        Dense_1024\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Serial[\n","          Serial[\n","            Serial[\n","              Branch_out3[\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","              ]\n","              DotProductCausalAttention_in3\n","              Serial[\n","                MergeHeads\n","              ]\n","              Dense_1024\n","            ]\n","          ]\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial_in3_out3[\n","    Branch_in3_out4[\n","      None\n","      Serial_in3_out3[\n","        LayerNorm\n","        Select[0,2,2,1,2]_in3_out5\n","        Serial_in4_out2[\n","          _in4_out4\n","          Serial_in4_out2[\n","            Parallel_in3_out3[\n","              Dense_1024\n","              Cache[\n","                Dense_1024\n","              ]\n","              Cache[\n","                Dense_1024\n","              ]\n","            ]\n","            PureAttention_in4_out2\n","            Dense_1024\n","          ]\n","          _in2_out2\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Dense_4096\n","        Serial[\n","          Relu\n","        ]\n","        Dropout\n","        Dense_1024\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Serial[\n","          Serial[\n","            Serial[\n","              Branch_out3[\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","              ]\n","              DotProductCausalAttention_in3\n","              Serial[\n","                MergeHeads\n","              ]\n","              Dense_1024\n","            ]\n","          ]\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial_in3_out3[\n","    Branch_in3_out4[\n","      None\n","      Serial_in3_out3[\n","        LayerNorm\n","        Select[0,2,2,1,2]_in3_out5\n","        Serial_in4_out2[\n","          _in4_out4\n","          Serial_in4_out2[\n","            Parallel_in3_out3[\n","              Dense_1024\n","              Cache[\n","                Dense_1024\n","              ]\n","              Cache[\n","                Dense_1024\n","              ]\n","            ]\n","            PureAttention_in4_out2\n","            Dense_1024\n","          ]\n","          _in2_out2\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Dense_4096\n","        Serial[\n","          Relu\n","        ]\n","        Dropout\n","        Dense_1024\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Serial[\n","          Serial[\n","            Serial[\n","              Branch_out3[\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","              ]\n","              DotProductCausalAttention_in3\n","              Serial[\n","                MergeHeads\n","              ]\n","              Dense_1024\n","            ]\n","          ]\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial_in3_out3[\n","    Branch_in3_out4[\n","      None\n","      Serial_in3_out3[\n","        LayerNorm\n","        Select[0,2,2,1,2]_in3_out5\n","        Serial_in4_out2[\n","          _in4_out4\n","          Serial_in4_out2[\n","            Parallel_in3_out3[\n","              Dense_1024\n","              Cache[\n","                Dense_1024\n","              ]\n","              Cache[\n","                Dense_1024\n","              ]\n","            ]\n","            PureAttention_in4_out2\n","            Dense_1024\n","          ]\n","          _in2_out2\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Dense_4096\n","        Serial[\n","          Relu\n","        ]\n","        Dropout\n","        Dense_1024\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Serial[\n","          Serial[\n","            Serial[\n","              Branch_out3[\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","              ]\n","              DotProductCausalAttention_in3\n","              Serial[\n","                MergeHeads\n","              ]\n","              Dense_1024\n","            ]\n","          ]\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial_in3_out3[\n","    Branch_in3_out4[\n","      None\n","      Serial_in3_out3[\n","        LayerNorm\n","        Select[0,2,2,1,2]_in3_out5\n","        Serial_in4_out2[\n","          _in4_out4\n","          Serial_in4_out2[\n","            Parallel_in3_out3[\n","              Dense_1024\n","              Cache[\n","                Dense_1024\n","              ]\n","              Cache[\n","                Dense_1024\n","              ]\n","            ]\n","            PureAttention_in4_out2\n","            Dense_1024\n","          ]\n","          _in2_out2\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Dense_4096\n","        Serial[\n","          Relu\n","        ]\n","        Dropout\n","        Dense_1024\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Serial[\n","          Serial[\n","            Serial[\n","              Branch_out3[\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","              ]\n","              DotProductCausalAttention_in3\n","              Serial[\n","                MergeHeads\n","              ]\n","              Dense_1024\n","            ]\n","          ]\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial_in3_out3[\n","    Branch_in3_out4[\n","      None\n","      Serial_in3_out3[\n","        LayerNorm\n","        Select[0,2,2,1,2]_in3_out5\n","        Serial_in4_out2[\n","          _in4_out4\n","          Serial_in4_out2[\n","            Parallel_in3_out3[\n","              Dense_1024\n","              Cache[\n","                Dense_1024\n","              ]\n","              Cache[\n","                Dense_1024\n","              ]\n","            ]\n","            PureAttention_in4_out2\n","            Dense_1024\n","          ]\n","          _in2_out2\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Dense_4096\n","        Serial[\n","          Relu\n","        ]\n","        Dropout\n","        Dense_1024\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Serial[\n","          Serial[\n","            Serial[\n","              Branch_out3[\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","              ]\n","              DotProductCausalAttention_in3\n","              Serial[\n","                MergeHeads\n","              ]\n","              Dense_1024\n","            ]\n","          ]\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial_in3_out3[\n","    Branch_in3_out4[\n","      None\n","      Serial_in3_out3[\n","        LayerNorm\n","        Select[0,2,2,1,2]_in3_out5\n","        Serial_in4_out2[\n","          _in4_out4\n","          Serial_in4_out2[\n","            Parallel_in3_out3[\n","              Dense_1024\n","              Cache[\n","                Dense_1024\n","              ]\n","              Cache[\n","                Dense_1024\n","              ]\n","            ]\n","            PureAttention_in4_out2\n","            Dense_1024\n","          ]\n","          _in2_out2\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Dense_4096\n","        Serial[\n","          Relu\n","        ]\n","        Dropout\n","        Dense_1024\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Serial[\n","          Serial[\n","            Serial[\n","              Branch_out3[\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","              ]\n","              DotProductCausalAttention_in3\n","              Serial[\n","                MergeHeads\n","              ]\n","              Dense_1024\n","            ]\n","          ]\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial_in3_out3[\n","    Branch_in3_out4[\n","      None\n","      Serial_in3_out3[\n","        LayerNorm\n","        Select[0,2,2,1,2]_in3_out5\n","        Serial_in4_out2[\n","          _in4_out4\n","          Serial_in4_out2[\n","            Parallel_in3_out3[\n","              Dense_1024\n","              Cache[\n","                Dense_1024\n","              ]\n","              Cache[\n","                Dense_1024\n","              ]\n","            ]\n","            PureAttention_in4_out2\n","            Dense_1024\n","          ]\n","          _in2_out2\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Dense_4096\n","        Serial[\n","          Relu\n","        ]\n","        Dropout\n","        Dense_1024\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Serial[\n","          Serial[\n","            Serial[\n","              Branch_out3[\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","              ]\n","              DotProductCausalAttention_in3\n","              Serial[\n","                MergeHeads\n","              ]\n","              Dense_1024\n","            ]\n","          ]\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial_in3_out3[\n","    Branch_in3_out4[\n","      None\n","      Serial_in3_out3[\n","        LayerNorm\n","        Select[0,2,2,1,2]_in3_out5\n","        Serial_in4_out2[\n","          _in4_out4\n","          Serial_in4_out2[\n","            Parallel_in3_out3[\n","              Dense_1024\n","              Cache[\n","                Dense_1024\n","              ]\n","              Cache[\n","                Dense_1024\n","              ]\n","            ]\n","            PureAttention_in4_out2\n","            Dense_1024\n","          ]\n","          _in2_out2\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Dense_4096\n","        Serial[\n","          Relu\n","        ]\n","        Dropout\n","        Dense_1024\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Serial[\n","          Serial[\n","            Serial[\n","              Branch_out3[\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","              ]\n","              DotProductCausalAttention_in3\n","              Serial[\n","                MergeHeads\n","              ]\n","              Dense_1024\n","            ]\n","          ]\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial_in3_out3[\n","    Branch_in3_out4[\n","      None\n","      Serial_in3_out3[\n","        LayerNorm\n","        Select[0,2,2,1,2]_in3_out5\n","        Serial_in4_out2[\n","          _in4_out4\n","          Serial_in4_out2[\n","            Parallel_in3_out3[\n","              Dense_1024\n","              Cache[\n","                Dense_1024\n","              ]\n","              Cache[\n","                Dense_1024\n","              ]\n","            ]\n","            PureAttention_in4_out2\n","            Dense_1024\n","          ]\n","          _in2_out2\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Dense_4096\n","        Serial[\n","          Relu\n","        ]\n","        Dropout\n","        Dense_1024\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Serial[\n","          Serial[\n","            Serial[\n","              Branch_out3[\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","              ]\n","              DotProductCausalAttention_in3\n","              Serial[\n","                MergeHeads\n","              ]\n","              Dense_1024\n","            ]\n","          ]\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial_in3_out3[\n","    Branch_in3_out4[\n","      None\n","      Serial_in3_out3[\n","        LayerNorm\n","        Select[0,2,2,1,2]_in3_out5\n","        Serial_in4_out2[\n","          _in4_out4\n","          Serial_in4_out2[\n","            Parallel_in3_out3[\n","              Dense_1024\n","              Cache[\n","                Dense_1024\n","              ]\n","              Cache[\n","                Dense_1024\n","              ]\n","            ]\n","            PureAttention_in4_out2\n","            Dense_1024\n","          ]\n","          _in2_out2\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Dense_4096\n","        Serial[\n","          Relu\n","        ]\n","        Dropout\n","        Dense_1024\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Serial[\n","          Serial[\n","            Serial[\n","              Branch_out3[\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","              ]\n","              DotProductCausalAttention_in3\n","              Serial[\n","                MergeHeads\n","              ]\n","              Dense_1024\n","            ]\n","          ]\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial_in3_out3[\n","    Branch_in3_out4[\n","      None\n","      Serial_in3_out3[\n","        LayerNorm\n","        Select[0,2,2,1,2]_in3_out5\n","        Serial_in4_out2[\n","          _in4_out4\n","          Serial_in4_out2[\n","            Parallel_in3_out3[\n","              Dense_1024\n","              Cache[\n","                Dense_1024\n","              ]\n","              Cache[\n","                Dense_1024\n","              ]\n","            ]\n","            PureAttention_in4_out2\n","            Dense_1024\n","          ]\n","          _in2_out2\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Dense_4096\n","        Serial[\n","          Relu\n","        ]\n","        Dropout\n","        Dense_1024\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Serial[\n","          Serial[\n","            Serial[\n","              Branch_out3[\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","              ]\n","              DotProductCausalAttention_in3\n","              Serial[\n","                MergeHeads\n","              ]\n","              Dense_1024\n","            ]\n","          ]\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial_in3_out3[\n","    Branch_in3_out4[\n","      None\n","      Serial_in3_out3[\n","        LayerNorm\n","        Select[0,2,2,1,2]_in3_out5\n","        Serial_in4_out2[\n","          _in4_out4\n","          Serial_in4_out2[\n","            Parallel_in3_out3[\n","              Dense_1024\n","              Cache[\n","                Dense_1024\n","              ]\n","              Cache[\n","                Dense_1024\n","              ]\n","            ]\n","            PureAttention_in4_out2\n","            Dense_1024\n","          ]\n","          _in2_out2\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Dense_4096\n","        Serial[\n","          Relu\n","        ]\n","        Dropout\n","        Dense_1024\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Serial[\n","          Serial[\n","            Serial[\n","              Branch_out3[\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","                [Dense_1024, Serial[\n","                  SplitIntoHeads\n","                ]]\n","              ]\n","              DotProductCausalAttention_in3\n","              Serial[\n","                MergeHeads\n","              ]\n","              Dense_1024\n","            ]\n","          ]\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial_in3_out3[\n","    Branch_in3_out4[\n","      None\n","      Serial_in3_out3[\n","        LayerNorm\n","        Select[0,2,2,1,2]_in3_out5\n","        Serial_in4_out2[\n","          _in4_out4\n","          Serial_in4_out2[\n","            Parallel_in3_out3[\n","              Dense_1024\n","              Cache[\n","                Dense_1024\n","              ]\n","              Cache[\n","                Dense_1024\n","              ]\n","            ]\n","            PureAttention_in4_out2\n","            Dense_1024\n","          ]\n","          _in2_out2\n","        ]\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  Serial[\n","    Branch_out2[\n","      None\n","      Serial[\n","        LayerNorm\n","        Dense_4096\n","        Serial[\n","          Relu\n","        ]\n","        Dropout\n","        Dense_1024\n","        Dropout\n","      ]\n","    ]\n","    Add_in2\n","  ]\n","  LayerNorm\n","  Select[0]_in3\n","  Dense_32000\n","]\n"],"name":"stdout"}]},{"cell_type":"code","metadata":{"id":"zzrxtGGDuUkc","executionInfo":{"status":"ok","timestamp":1624414022161,"user_tz":-420,"elapsed":454,"user":{"displayName":"Quý Trần Nhật","photoUrl":"","userId":"13689480600881371104"}}},"source":["# create inputs\n","# a simple example \n","# inputs = 'question: She asked him where is john? context: John was at the game'\n","\n","# an extensive example\n","inputs = 'question: What are some of the colours of a rose? context: A rose is a woody perennial flowering plant of the genus Rosa, in the family Rosaceae, or the flower it bears.There are over three hundred species and tens of thousands of cultivars. They form a group of plants that can be erect shrubs, climbing, or trailing, with stems that are often armed with sharp prickles. Flowers vary in size and shape and are usually large and showy, in colours ranging from white through yellows and reds. Most species are native to Asia, with smaller numbers native to Europe, North America, and northwestern Africa. Species, cultivars and hybrids are all widely grown for their beauty and often are fragrant.'"],"execution_count":12,"outputs":[]},{"cell_type":"code","metadata":{"id":"nVlu-iwHcZRY","colab":{"base_uri":"https://localhost:8080/"},"executionInfo":{"status":"ok","timestamp":1624414023745,"user_tz":-420,"elapsed":458,"user":{"displayName":"Quý Trần Nhật","photoUrl":"","userId":"13689480600881371104"}},"outputId":"a5c2bad0-e948-44dd-b00f-e96330ab3ab4"},"source":["# tokenizing the input so we could feed it for decoding\n","print(tokenize(inputs))\n","test_inputs = tokenize(inputs) "],"execution_count":13,"outputs":[{"output_type":"stream","text":["[  822    10   363    33   128    13     8  6548    13     3     9  4659\n","    58  2625    10    71  4659    19     3     9  1679    63 24999  5624\n","    53  1475    13     8     3   729   302 15641     6    16     8   384\n"," 15641  8433    15     6    42     8  5624    34  4595     7     5  7238\n","    33   147   386  6189  3244    11     3   324     7    13  2909    13\n"," 10357   291     7     5   328   607     3     9   563    13  2677    24\n","    54    36     3    15 12621 21675     7     6 11908     6    42  5032\n","    53     6    28  6269     7    24    33   557     3  8715    28  4816\n","     3  2246 19376     7     5 20294  5215    16   812    11  2346    11\n","    33  1086   508    11   504    63     6    16  6548     3  6836    45\n","   872   190  4459     7    11  1131     7     5  1377  3244    33  4262\n","    12  3826     6    28  2755  2302  4262    12  1740     6  1117  1371\n","     6    11  3457 24411  2648     5     3  7727   725     6 10357   291\n","     7    11  9279     7    33    66  5456  4503    21    70  2790    11\n","   557    33 29346     5]\n"],"name":"stdout"}]},{"cell_type":"code","metadata":{"id":"PJh_Qw9G5jND","colab":{"base_uri":"https://localhost:8080/"},"executionInfo":{"status":"ok","timestamp":1624414093574,"user_tz":-420,"elapsed":65167,"user":{"displayName":"Quý Trần Nhật","photoUrl":"","userId":"13689480600881371104"}},"outputId":"8e00c30f-9e58-4b5b-cb5c-ae1a73066c97"},"source":["# Temperature is a parameter for sampling.\n","#   # * 0.0: same as argmax, always pick the most probable token\n","#   # * 1.0: sampling from the distribution (can sometimes say random things)\n","#   # * values inbetween can trade off diversity and quality, try it out!\n","output = decoding.autoregressive_sample(model, inputs=np.array(test_inputs)[None, :],\n","                                        temperature=0.0, max_length=10)\n","print(wrapper.fill(pretty_decode(output[0])))"],"execution_count":14,"outputs":[{"output_type":"stream","text":["white through yellows and reds\n"],"name":"stdout"}]},{"cell_type":"markdown","metadata":{"id":"BadBty9C-8wY"},"source":["### Note: As you can see the RAM is almost full, it is because the model and the decoding is memory heavy. You can run decoding just once. Running it the second time with another example might give you the same answer as before, or not run at all (crash). If that happens restart the runtime (see how to at the start of the notebook) and run all the cells again."]}]}