docs/notebooks/talk2knowledgegraphs/tutorial_starkqa_primekg_loader.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Tutorial: StarkQA-PrimeKG Loader"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "In this tutorial, we will explain how to load StarkQA-PrimeKG dataset, which is a dataset for question answering over knowledge graphs of PrimeKG. \n",
    "\n",
    "Prior information about the StarkQA-PrimeKG dataaset can be found in the following repositories:\n",
    "- https://github.com/snap-stanford/stark\n",
    "- https://stark.stanford.edu/\n",
    "- https://huggingface.co/datasets/snap-stanford/stark"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "We first need to import the necessary libraries as follows."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "c:\\Users\\mulyadi\\Repo\\AIAgents4Pharma\\venv\\Lib\\site-packages\\pydantic\\_internal\\_fields.py:132: UserWarning: Field \"model_id\" in SysBioModel has conflict with protected namespace \"model_\".\n",
      "\n",
      "You may be able to resolve this warning by setting `model_config['protected_namespaces'] = ()`.\n",
      "  warnings.warn(\n",
      "c:\\Users\\mulyadi\\Repo\\AIAgents4Pharma\\venv\\Lib\\site-packages\\pydantic\\_internal\\_fields.py:132: UserWarning: Field \"model_id\" in BasicoModel has conflict with protected namespace \"model_\".\n",
      "\n",
      "You may be able to resolve this warning by setting `model_config['protected_namespaces'] = ()`.\n",
      "  warnings.warn(\n",
      "c:\\Users\\mulyadi\\Repo\\AIAgents4Pharma\\venv\\Lib\\site-packages\\pydantic\\_internal\\_fields.py:132: UserWarning: Field \"model_data\" in SimulateModelInput has conflict with protected namespace \"model_\".\n",
      "\n",
      "You may be able to resolve this warning by setting `model_config['protected_namespaces'] = ()`.\n",
      "  warnings.warn(\n"
     ]
    }
   ],
   "source": [
    "# Import necessary libraries\n",
    "import sys\n",
    "sys.path.append('../../..')\n",
    "from aiagents4pharma.talk2knowledgegraphs.datasets.starkqa_primekg import StarkQAPrimeKG"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Load StarQA-PrimeKG"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "The `StarkQAPrimeKG` allows to load the data from the HuggingFace Hub if the data is not available locally. \n",
    "\n",
    "Otherwise, the data is loaded from the local directory as defined in the `local_dir`."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Define starkqa primekg data by providing a local directory where the data is stored\n",
    "starkqa_data = StarkQAPrimeKG(local_dir=\"../../../../data/starkqa_primekg_test/\")"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "To load the dataframes of StarkQA and its split, we just need a method as follows."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Loading StarkQAPrimeKG dataset...\n",
      "Downloading files from snap-stanford/stark\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|          | 0/7 [00:00<?, ?it/s]"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "cc0a47ce0d12479788efd603a00f5df4",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "qa/prime/split/test-0.1.index:   0%|          | 0.00/1.39k [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      " 14%|█▍        | 1/7 [00:00<00:02,  2.64it/s]"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "77d6023aa4f846688bba1f1d451f0fed",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "qa/prime/split/test.index:   0%|          | 0.00/14.0k [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      " 29%|██▊       | 2/7 [00:00<00:02,  2.49it/s]"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "0d4cee03cd6e47fb8625d2d3f756b4f5",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "qa/prime/split/train.index:   0%|          | 0.00/30.9k [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      " 43%|████▎     | 3/7 [00:01<00:01,  2.61it/s]"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "f8d4d41ff6f048f09cbcb527c3ba0cd9",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "qa/prime/split/val.index:   0%|          | 0.00/11.2k [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      " 57%|█████▋    | 4/7 [00:01<00:01,  2.02it/s]"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "423b6fbd46434edf8856e5cb1f07c6b1",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "qa/prime/stark_qa/stark_qa.csv:   0%|          | 0.00/2.06M [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      " 71%|███████▏  | 5/7 [00:02<00:01,  1.65it/s]"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "29cca8a2dd2046a6b9427bae3258fa14",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "(…)ark_qa/stark_qa_human_generated_eval.csv:   0%|          | 0.00/14.8k [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      " 86%|████████▌ | 6/7 [00:02<00:00,  1.93it/s]"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "4fa7c0ca1ef04f9a901b2cdef7d2c744",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "processed.zip:   0%|          | 0.00/28.0M [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "100%|██████████| 7/7 [00:04<00:00,  1.54it/s]\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Loading StarkQAPrimeKG embeddings...\n",
      "Downloading embeddings from text-embedding-ada-002\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "Downloading...\n",
      "From (original): https://drive.google.com/uc?id=1MshwJttPZsHEM2cKA5T13SIrsLeBEdyU\n",
      "From (redirected): https://drive.google.com/uc?id=1MshwJttPZsHEM2cKA5T13SIrsLeBEdyU&confirm=t&uuid=ae35d8f3-b8d6-454d-bb67-f8d73223508d\n",
      "To: c:\\Users\\mulyadi\\Repo\\data\\starkqa_primekg_test\\text-embedding-ada-002\\query\\query_emb_dict.pt\n",
      "100%|██████████| 72.0M/72.0M [00:02<00:00, 31.3MB/s]\n",
      "Downloading...\n",
      "From (original): https://drive.google.com/uc?id=16EJvCMbgkVrQ0BuIBvLBp-BYPaye-Edy\n",
      "From (redirected): https://drive.google.com/uc?id=16EJvCMbgkVrQ0BuIBvLBp-BYPaye-Edy&confirm=t&uuid=a9359c11-415b-40f2-a72e-ab25fcd5372b\n",
      "To: c:\\Users\\mulyadi\\Repo\\data\\starkqa_primekg_test\\text-embedding-ada-002\\doc\\candidate_emb_dict.pt\n",
      "100%|██████████| 832M/832M [00:34<00:00, 24.1MB/s] \n",
      "c:\\Users\\mulyadi\\Repo\\AIAgents4Pharma\\docs\\notebooks\\talk2knowledgegraphs\\../../..\\aiagents4pharma\\talk2knowledgegraphs\\datasets\\starkqa_primekg.py:140: FutureWarning: You are using `torch.load` with `weights_only=False` (the current default value), which uses the default pickle module implicitly. It is possible to construct malicious pickle data which will execute arbitrary code during unpickling (See https://github.com/pytorch/pytorch/blob/main/SECURITY.md#untrusted-models for more details). In a future release, the default value for `weights_only` will be flipped to `True`. This limits the functions that could be executed during unpickling. Arbitrary objects will no longer be allowed to be loaded via this mode unless they are explicitly allowlisted by the user via `torch.serialization.add_safe_globals`. We recommend you start setting `weights_only=True` for any use case where you don't have full control of the loaded file. Please open an issue on GitHub for any issues related to this experimental feature.\n",
      "  query_emb_dict = torch.load(query_emb_path)\n",
      "c:\\Users\\mulyadi\\Repo\\AIAgents4Pharma\\docs\\notebooks\\talk2knowledgegraphs\\../../..\\aiagents4pharma\\talk2knowledgegraphs\\datasets\\starkqa_primekg.py:141: FutureWarning: You are using `torch.load` with `weights_only=False` (the current default value), which uses the default pickle module implicitly. It is possible to construct malicious pickle data which will execute arbitrary code during unpickling (See https://github.com/pytorch/pytorch/blob/main/SECURITY.md#untrusted-models for more details). In a future release, the default value for `weights_only` will be flipped to `True`. This limits the functions that could be executed during unpickling. Arbitrary objects will no longer be allowed to be loaded via this mode unless they are explicitly allowlisted by the user via `torch.serialization.add_safe_globals`. We recommend you start setting `weights_only=True` for any use case where you don't have full control of the loaded file. Please open an issue on GitHub for any issues related to this experimental feature.\n",
      "  node_emb_dict = torch.load(node_emb_path)\n"
     ]
    }
   ],
   "source": [
    "# Invoke a method to load the data\n",
    "starkqa_data.load_data()\n",
    "\n",
    "# Get the StarkQAPrimeKG data, which are the QA pairs, split indices, and the node information\n",
    "starkqa_df = starkqa_data.get_starkqa()\n",
    "starkqa_split_indices = starkqa_data.get_starkqa_split_indicies()\n",
    "starkqa_node_info = starkqa_data.get_starkqa_node_info()\n",
    "query_embeddings = starkqa_data.get_query_embeddings()\n",
    "node_embeddings = starkqa_data.get_node_embeddings()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Check StarQA-PrimeKG Dataframes"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "\n",
    "StarkQA dataframes contain the following columns:\n",
    "- `id`: Unique identifier for each question and answer pair\n",
    "- `query`: The synthesized question from the StarkQA dataset\n",
    "- `answer_ids`: The unique identifier for the answer to the question (multiple answers are possible)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>id</th>\n",
       "      <th>query</th>\n",
       "      <th>answer_ids</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>0</td>\n",
       "      <td>Could you identify any skin diseases associate...</td>\n",
       "      <td>[95886]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>1</td>\n",
       "      <td>What drugs target the CYP3A4 enzyme and are us...</td>\n",
       "      <td>[15450]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>2</td>\n",
       "      <td>What is the name of the condition characterize...</td>\n",
       "      <td>[98851, 98853]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>3</td>\n",
       "      <td>What drugs are used to treat epithelioid sarco...</td>\n",
       "      <td>[15698]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>4</td>\n",
       "      <td>Can you supply a compilation of genes and prot...</td>\n",
       "      <td>[7161, 22045]</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   id                                              query      answer_ids\n",
       "0   0  Could you identify any skin diseases associate...         [95886]\n",
       "1   1  What drugs target the CYP3A4 enzyme and are us...         [15450]\n",
       "2   2  What is the name of the condition characterize...  [98851, 98853]\n",
       "3   3  What drugs are used to treat epithelioid sarco...         [15698]\n",
       "4   4  Can you supply a compilation of genes and prot...   [7161, 22045]"
      ]
     },
     "execution_count": 5,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Check a sample of the starkqa primekg dataframe\n",
    "starkqa_df.head()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "The current StarkQA-PrimeKG has about 11K records of questions and answers pairs."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "(11204, 3)"
      ]
     },
     "execution_count": 6,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Check dimensions of the starkqa primekg dataframe\n",
    "starkqa_df.shape"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### CHeck StarkQA-PrimeKG Node Information"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "StarkQA provides an additional node information for PrimeKG as a dictionary for each node.\n",
    "\n",
    "This allows us to further enrich the features of the knowledge graph nodes."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'id': 9796,\n",
       " 'type': 'gene/protein',\n",
       " 'name': 'PHYHIP',\n",
       " 'source': 'NCBI',\n",
       " 'details': {'query': 'PHYHIP',\n",
       "  '_id': '9796',\n",
       "  '_score': 17.934021,\n",
       "  'alias': ['DYRK1AP3', 'PAHX-AP', 'PAHXAP1'],\n",
       "  'genomic_pos': {'chr': '8',\n",
       "   'end': 22232101,\n",
       "   'ensemblgene': 'ENSG00000168490',\n",
       "   'start': 22219703,\n",
       "   'strand': -1},\n",
       "  'name': 'phytanoyl-CoA 2-hydroxylase interacting protein',\n",
       "  'summary': 'Enables protein tyrosine kinase binding activity. Involved in protein localization. Located in cytoplasm. [provided by Alliance of Genome Resources, Apr 2022]'}}"
      ]
     },
     "execution_count": 7,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Check the node information of PrimeKG\n",
    "starkqa_node_info[0]"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Check StarQA-PrimeKG Embeddings (Query & Nodes)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Note that StarkQA has provided pre-loaded embeddings for the queries and nodes using 'text-embedding-ada-002' model. "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{3461: tensor([[-0.0211, -0.0149,  0.0020,  ..., -0.0277, -0.0085, -0.0034]]),\n",
       " 10065: tensor([[-0.0131,  0.0068,  0.0197,  ..., -0.0382, -0.0209, -0.0300]]),\n",
       " 8931: tensor([[-0.0299, -0.0173,  0.0056,  ..., -0.0206, -0.0150, -0.0108]]),\n",
       " 6664: tensor([[-0.0368,  0.0045,  0.0026,  ..., -0.0055, -0.0094, -0.0178]]),\n",
       " 3798: tensor([[ 0.0009, -0.0095,  0.0231,  ..., -0.0197, -0.0106, -0.0288]]),\n",
       " 9392: tensor([[-0.0092,  0.0213,  0.0097,  ..., -0.0140, -0.0023, -0.0223]]),\n",
       " 1039: tensor([[-0.0093, -0.0078,  0.0324,  ..., -0.0046, -0.0062, -0.0395]]),\n",
       " 2116: tensor([[-0.0197, -0.0117,  0.0251,  ..., -0.0249, -0.0022, -0.0175]]),\n",
       " 930: tensor([[-0.0028, -0.0036,  0.0514,  ..., -0.0220, -0.0057, -0.0295]]),\n",
       " 7313: tensor([[-0.0350, -0.0214, -0.0035,  ..., -0.0162, -0.0054, -0.0175]]),\n",
       " 8787: tensor([[-0.0149, -0.0030, -0.0097,  ..., -0.0245,  0.0041, -0.0193]]),\n",
       " 7271: tensor([[-0.0099, -0.0031, -0.0148,  ..., -0.0317, -0.0035, -0.0182]]),\n",
       " 7188: tensor([[-0.0228, -0.0016,  0.0180,  ..., -0.0207, -0.0164, -0.0222]]),\n",
       " 852: tensor([[ 0.0112,  0.0029,  0.0024,  ..., -0.0040, -0.0215, -0.0425]]),\n",
       " 2074: tensor([[-0.0051,  0.0221,  0.0356,  ..., -0.0125, -0.0135, -0.0233]]),\n",
       " 6706: tensor([[-0.0394,  0.0032,  0.0088,  ..., -0.0093, -0.0176, -0.0242]]),\n",
       " 8853: tensor([[-0.0057, -0.0016,  0.0161,  ..., -0.0451, -0.0100, -0.0232]]),\n",
       " 10107: tensor([[-0.0087,  0.0026,  0.0312,  ..., -0.0094, -0.0063, -0.0216]]),\n",
       " 3503: tensor([[-0.0094, -0.0015,  0.0218,  ..., -0.0126,  0.0032, -0.0342]]),\n",
       " 9109: tensor([[-1.4699e-02, -1.1998e-02,  2.8852e-02,  ..., -1.6036e-02,\n",
       "          -8.6785e-05, -1.4340e-03]]),\n",
       " 5629: tensor([[-0.0078, -0.0197,  0.0213,  ..., -0.0316, -0.0101, -0.0270]]),\n",
       " 786: tensor([[-0.0166, -0.0058,  0.0396,  ..., -0.0295, -0.0308, -0.0264]]),\n",
       " 2149: tensor([[-0.0022, -0.0129,  0.0103,  ..., -0.0195, -0.0208, -0.0121]]),\n",
       " 4263: tensor([[-0.0005,  0.0076,  0.0375,  ..., -0.0059, -0.0061, -0.0149]]),\n",
       " 8543: tensor([[-0.0281, -0.0157,  0.0122,  ..., -0.0255, -0.0268, -0.0206]]),\n",
       " 3813: tensor([[ 0.0065,  0.0089,  0.0253,  ..., -0.0556, -0.0098, -0.0037]]),\n",
       " 1066: tensor([[-0.0221,  0.0203,  0.0152,  ...,  0.0065,  0.0005, -0.0114]]),\n",
       " 5714: tensor([[-0.0168,  0.0073,  0.0284,  ..., -0.0201, -0.0329, -0.0017]]),\n",
       " 9034: tensor([[-0.0095,  0.0073, -0.0046,  ..., -0.0111, -0.0089, -0.0426]]),\n",
       " 542: tensor([[ 0.0021,  0.0020,  0.0346,  ...,  0.0011, -0.0102, -0.0250]]),\n",
       " 420: tensor([[-0.0139,  0.0170,  0.0135,  ..., -0.0293, -0.0088, -0.0302]]),\n",
       " 5676: tensor([[-0.0170, -0.0104,  0.0342,  ..., -0.0181, -0.0195, -0.0036]]),\n",
       " 9156: tensor([[-0.0200,  0.0027,  0.0147,  ..., -0.0106, -0.0115, -0.0181]]),\n",
       " 10158: tensor([[-0.0252,  0.0022,  0.0043,  ..., -0.0211, -0.0061, -0.0353]]),\n",
       " 6759: tensor([[-0.0093,  0.0119,  0.0113,  ..., -0.0054, -0.0099, -0.0212]]),\n",
       " 1104: tensor([[-0.0240,  0.0008,  0.0275,  ..., -0.0006, -0.0232, -0.0160]]),\n",
       " 3971: tensor([[-0.0501,  0.0005,  0.0088,  ..., -0.0182, -0.0297, -0.0123]]),\n",
       " 4301: tensor([[-0.0137,  0.0176,  0.0173,  ..., -0.0139, -0.0050, -0.0388]]),\n",
       " 8421: tensor([[-0.0127,  0.0035,  0.0169,  ..., -0.0300, -0.0069, -0.0168]]),\n",
       " 1368: tensor([[-0.0133, -0.0069,  0.0375,  ..., -0.0279, -0.0215, -0.0264]]),\n",
       " 4294: tensor([[-0.0056,  0.0164,  0.0043,  ..., -0.0308, -0.0014, -0.0142]]),\n",
       " 6918: tensor([[-0.0028, -0.0118,  0.0264,  ..., -0.0316, -0.0020, -0.0116]]),\n",
       " 1091: tensor([[ 0.0121,  0.0109,  0.0113,  ..., -0.0101, -0.0160, -0.0149]]),\n",
       " 2247: tensor([[-0.0230, -0.0102,  0.0272,  ..., -0.0018, -0.0046, -0.0406]]),\n",
       " 5837: tensor([[-0.0192,  0.0025,  0.0138,  ...,  0.0058, -0.0295, -0.0283]]),\n",
       " 7042: tensor([[-0.0333, -0.0113,  0.0064,  ..., -0.0020,  0.0134, -0.0209]]),\n",
       " 998: tensor([[-0.0104,  0.0054,  0.0243,  ..., -0.0125, -0.0161, -0.0089]]),\n",
       " 3730: tensor([[-0.0061, -0.0037,  0.0204,  ..., -0.0268, -0.0134, -0.0258]]),\n",
       " 8999: tensor([[-0.0265,  0.0045,  0.0028,  ..., -0.0303, -0.0129, -0.0295]]),\n",
       " 10334: tensor([[-0.0227,  0.0155,  0.0216,  ..., -0.0149, -0.0134, -0.0084]]),\n",
       " 6535: tensor([[-0.0200,  0.0091,  0.0158,  ...,  0.0030,  0.0076, -0.0229]]),\n",
       " 6457: tensor([[-0.0086,  0.0287,  0.0493,  ..., -0.0138, -0.0232, -0.0324]]),\n",
       " 10256: tensor([[-0.0432,  0.0020, -0.0082,  ..., -0.0145,  0.0104, -0.0290]]),\n",
       " 3652: tensor([[ 0.0045, -0.0083,  0.0340,  ..., -0.0242, -0.0020, -0.0075]]),\n",
       " 5578: tensor([[-0.0149,  0.0094,  0.0102,  ..., -0.0338, -0.0161, -0.0152]]),\n",
       " 9258: tensor([[-0.0296,  0.0062,  0.0231,  ..., -0.0346, -0.0119, -0.0425]]),\n",
       " 5681: tensor([[-0.0175, -0.0027,  0.0320,  ..., -0.0204, -0.0251, -0.0168]]),\n",
       " 3986: tensor([[ 0.0079,  0.0110,  0.0342,  ..., -0.0180, -0.0265, -0.0248]]),\n",
       " 7120: tensor([[-0.0079, -0.0056,  0.0072,  ..., -0.0260, -0.0142, -0.0088]]),\n",
       " 5955: tensor([[-0.0173, -0.0054,  0.0010,  ..., -0.0134, -0.0200, -0.0200]]),\n",
       " 2325: tensor([[-0.0170,  0.0122,  0.0286,  ..., -0.0136, -0.0165, -0.0331]]),\n",
       " 9365: tensor([[-0.0132, -0.0187,  0.0430,  ..., -0.0203,  0.0008, -0.0165]]),\n",
       " 5445: tensor([[-0.0329,  0.0022,  0.0292,  ...,  0.0036, -0.0263, -0.0159]]),\n",
       " 613: tensor([[-0.0187,  0.0050,  0.0191,  ..., -0.0364, -0.0244, -0.0135]]),\n",
       " 3496: tensor([[-0.0220, -0.0019,  0.0207,  ..., -0.0054, -0.0342, -0.0121]]),\n",
       " 10092: tensor([[-0.0175,  0.0183,  0.0100,  ..., -0.0265, -0.0226, -0.0040]]),\n",
       " 6693: tensor([[-0.0317,  0.0198,  0.0161,  ..., -0.0102, -0.0163, -0.0229]]),\n",
       " 5868: tensor([[-0.0262, -0.0053,  0.0163,  ...,  0.0011, -0.0109, -0.0221]]),\n",
       " 2218: tensor([[-0.0202,  0.0133, -0.0045,  ..., -0.0187, -0.0169, -0.0417]]),\n",
       " 8612: tensor([[-0.0237,  0.0046,  0.0122,  ..., -0.0015, -0.0058, -0.0221]]),\n",
       " 4132: tensor([[-0.0258, -0.0105,  0.0017,  ..., -0.0209, -0.0006, -0.0162]]),\n",
       " 6947: tensor([[-0.0323,  0.0094,  0.0045,  ..., -0.0046, -0.0078, -0.0291]]),\n",
       " 1337: tensor([[-0.0010, -0.0075,  0.0179,  ..., -0.0258, -0.0006, -0.0082]]),\n",
       " 7286: tensor([[-0.0201,  0.0169,  0.0172,  ..., -0.0247, -0.0228, -0.0347]]),\n",
       " 2083: tensor([[-0.0234,  0.0181,  0.0052,  ..., -0.0534, -0.0115,  0.0063]]),\n",
       " 8489: tensor([[-0.0047,  0.0084,  0.0338,  ..., -0.0213,  0.0007, -0.0218]]),\n",
       " 1255: tensor([[ 0.0202,  0.0071,  0.0189,  ..., -0.0255, -0.0096, -0.0177]]),\n",
       " 6825: tensor([[-0.0191, -0.0261,  0.0079,  ..., -0.0010, -0.0084, -0.0248]]),\n",
       " 8770: tensor([[-0.0163,  0.0037,  0.0011,  ..., -0.0142, -0.0016, -0.0057]]),\n",
       " 4050: tensor([[-0.0149, -0.0078,  0.0230,  ..., -0.0075,  0.0090,  0.0053]]),\n",
       " 771: tensor([[-0.0440, -0.0155,  0.0063,  ..., -0.0161, -0.0211, -0.0219]]),\n",
       " 488: tensor([[-0.0064,  0.0228,  0.0214,  ..., -0.0275, -0.0211,  0.0056]]),\n",
       " 9207: tensor([[-0.0238, -0.0228,  0.0213,  ..., -0.0002, -0.0113, -0.0265]]),\n",
       " 5527: tensor([[-0.0267,  0.0013,  0.0085,  ..., -0.0055, -0.0107, -0.0045]]),\n",
       " 10209: tensor([[-0.0412, -0.0122,  0.0052,  ..., -0.0177, -0.0107, -0.0370]]),\n",
       " 6408: tensor([[ 0.0111,  0.0030,  0.0354,  ..., -0.0328, -0.0027, -0.0214]]),\n",
       " 10871: tensor([[-0.0222, -0.0064,  0.0159,  ..., -0.0176, -0.0043, -0.0115]]),\n",
       " 1400: tensor([[-0.0196, -0.0125, -0.0099,  ..., -0.0028, -0.0108, -0.0127]]),\n",
       " 4605: tensor([[-0.0068,  0.0140,  0.0300,  ..., -0.0113, -0.0175, -0.0164]]),\n",
       " 8125: tensor([[-0.0077,  0.0130,  0.0203,  ..., -0.0057, -0.0104, -0.0366]]),\n",
       " 3058: tensor([[-0.0173, -0.0141,  0.0387,  ..., -0.0130, -0.0177, -0.0134]]),\n",
       " 2902: tensor([[-0.0136,  0.0150, -0.0038,  ..., -0.0203, -0.0187, -0.0287]]),\n",
       " 124: tensor([[-0.0468, -0.0133,  0.0058,  ..., -0.0092, -0.0075, -0.0094]]),\n",
       " 5372: tensor([[-0.0283,  0.0090,  0.0270,  ..., -0.0121, -0.0208, -0.0152]]),\n",
       " 9452: tensor([[-0.0077, -0.0090,  0.0313,  ...,  0.0045, -0.0063, -0.0306]]),\n",
       " 5210: tensor([[-0.0040, -0.0021, -0.0022,  ..., -0.0374, -0.0116, -0.0267]]),\n",
       " 9530: tensor([[-0.0135,  0.0100, -0.0026,  ..., -0.0265, -0.0167, -0.0276]]),\n",
       " 2860: tensor([[-0.0281,  0.0047,  0.0127,  ..., -0.0035, -0.0248, -0.0326]]),\n",
       " 4767: tensor([[-0.0040, -0.0116,  0.0258,  ..., -0.0258, -0.0098, -0.0208]]),\n",
       " 8047: tensor([[-0.0243,  0.0211,  0.0211,  ..., -0.0145, -0.0168, -0.0290]]),\n",
       " 10913: tensor([[-0.0265,  0.0101,  0.0201,  ..., -0.0030, -0.0126, -0.0202]]),\n",
       " 1562: tensor([[-0.0129,  0.0169,  0.0356,  ..., -0.0103, -0.0202, -0.0375]]),\n",
       " 7648: tensor([[-0.0453, -0.0020, -0.0037,  ..., -0.0268, -0.0088, -0.0078]]),\n",
       " 11049: tensor([[-0.0303, -0.0067,  0.0083,  ..., -0.0103, -0.0154, -0.0345]]),\n",
       " 282: tensor([[-0.0283, -0.0122,  0.0196,  ..., -0.0083, -0.0086, -0.0130]]),\n",
       " 6202: tensor([[-0.0193,  0.0041,  0.0195,  ..., -0.0260,  0.0192, -0.0293]]),\n",
       " 3007: tensor([[-0.0128,  0.0127,  0.0187,  ..., -0.0271, -0.0209, -0.0299]]),\n",
       " 1872: tensor([[-0.0277,  0.0166,  0.0232,  ..., -0.0356, -0.0066, -0.0306]]),\n",
       " 10403: tensor([[-0.0248,  0.0171,  0.0214,  ..., -0.0203, -0.0136, -0.0122]]),\n",
       " 9820: tensor([[-0.0175, -0.0096,  0.0098,  ..., -0.0108, -0.0033, -0.0320]]),\n",
       " 7775: tensor([[-0.0310, -0.0080,  0.0322,  ..., -0.0170, -0.0141, -0.0167]]),\n",
       " 2689: tensor([[-0.0079,  0.0006,  0.0027,  ..., -0.0296, -0.0027, -0.0134]]),\n",
       " 2570: tensor([[-0.0048, -0.0163,  0.0272,  ..., -0.0294, -0.0164, -0.0257]]),\n",
       " 11174: tensor([[-0.0170, -0.0050,  0.0291,  ..., -0.0246,  0.0023, -0.0130]]),\n",
       " 8283: tensor([[-0.0211, -0.0116,  0.0121,  ..., -0.0151, -0.0055, -0.0515]]),\n",
       " 11016: tensor([[-0.0458, -0.0117,  0.0123,  ..., -0.0167, -0.0029, -0.0164]]),\n",
       " 2412: tensor([[ 0.0082, -0.0206,  0.0084,  ..., -0.0079, -0.0215, -0.0158]]),\n",
       " 7617: tensor([[ 0.0304,  0.0039,  0.0335,  ..., -0.0297,  0.0074, -0.0147]]),\n",
       " 9942: tensor([[-0.0142, -0.0155,  0.0257,  ..., -0.0116, -0.0039, -0.0455]]),\n",
       " 8018: tensor([[-0.0044, -0.0004,  0.0337,  ..., -0.0171, -0.0061, -0.0022]]),\n",
       " 4738: tensor([[-0.0286, -0.0077,  0.0306,  ...,  0.0021, -0.0156, -0.0221]]),\n",
       " 9696: tensor([[-0.0330,  0.0072,  0.0208,  ..., -0.0230, -0.0036, -0.0235]]),\n",
       " 1910: tensor([[ 0.0014, -0.0117,  0.0504,  ..., -0.0166, -0.0124, -0.0140]]),\n",
       " 10561: tensor([[ 0.0044,  0.0262,  0.0258,  ..., -0.0260, -0.0152, -0.0078]]),\n",
       " 6099: tensor([[-0.0217,  0.0061,  0.0240,  ..., -0.0234,  0.0041, -0.0285]]),\n",
       " 3165: tensor([[-0.0137, -0.0045, -0.0006,  ..., -0.0238, -0.0121, -0.0204]]),\n",
       " 6360: tensor([[-0.0304, -0.0149,  0.0262,  ..., -0.0192, -0.0082, -0.0070]]),\n",
       " 10698: tensor([[-0.0263, -0.0104,  0.0114,  ..., -0.0173, -0.0208, -0.0281]]),\n",
       " 3309: tensor([[-0.0039, -0.0026,  0.0199,  ..., -0.0197, -0.0355, -0.0057]]),\n",
       " 4979: tensor([[-0.0032,  0.0127,  0.0098,  ..., -0.0219, -0.0092, -0.0433]]),\n",
       " 1885: tensor([[-0.0315, -0.0143,  0.0015,  ..., -0.0256, -0.0130, -0.0251]]),\n",
       " 275: tensor([[-0.0173,  0.0086, -0.0024,  ..., -0.0134, -0.0171, -0.0136]]),\n",
       " 9703: tensor([[-0.0395, -0.0045,  0.0138,  ..., -0.0100, -0.0152, -0.0392]]),\n",
       " 5023: tensor([[-0.0242,  0.0085,  0.0349,  ..., -0.0191, -0.0232, -0.0368]]),\n",
       " 7856: tensor([[-0.0409,  0.0098,  0.0180,  ...,  0.0030, -0.0007, -0.0336]]),\n",
       " 1751: tensor([[-0.0105, -0.0003,  0.0147,  ..., -0.0061, -0.0254, -0.0027]]),\n",
       " 8274: tensor([[ 0.0091, -0.0159,  0.0305,  ..., -0.0159, -0.0164,  0.0019]]),\n",
       " 4554: tensor([[-0.0300, -0.0127,  0.0328,  ..., -0.0587, -0.0193, -0.0246]]),\n",
       " 7782: tensor([[-0.0098,  0.0006,  0.0294,  ..., -0.0161, -0.0156, -0.0176]]),\n",
       " 2587: tensor([[-0.0019, -0.0106,  0.0182,  ..., -0.0078, -0.0100, -0.0259]]),\n",
       " 11183: tensor([[-0.0030, -0.0023,  0.0406,  ..., -0.0128, -0.0119, -0.0280]]),\n",
       " 8316: tensor([[-0.0138, -0.0070,  0.0182,  ..., -0.0271, -0.0366, -0.0164]]),\n",
       " 4436: tensor([[-0.0424, -0.0058, -0.0143,  ..., -0.0308, -0.0188,  0.0010]]),\n",
       " 1633: tensor([[-0.0193,  0.0028,  0.0055,  ..., -0.0162, -0.0030, -0.0049]]),\n",
       " 22: tensor([[-0.0268,  0.0052,  0.0068,  ..., -0.0240,  0.0091, -0.0172]]),\n",
       " 7519: tensor([[-0.0226,  0.0042,  0.0367,  ..., -0.0293,  0.0084, -0.0331]]),\n",
       " 10596: tensor([[-0.0290,  0.0010,  0.0177,  ..., -0.0268, -0.0102, -0.0267]]),\n",
       " 3192: tensor([[-0.0218,  0.0029,  0.0183,  ...,  0.0242, -0.0030, -0.0400]]),\n",
       " 6397: tensor([[-0.0131, -0.0062,  0.0424,  ..., -0.0310, -0.0192, -0.0157]]),\n",
       " 7934: tensor([[-0.0252, -0.0025,  0.0182,  ..., -0.0268, -0.0074, -0.0292]]),\n",
       " 9661: tensor([[-0.0240, -0.0007, -0.0016,  ..., -0.0156, -0.0142, -0.0245]]),\n",
       " 5141: tensor([[-0.0020,  0.0147,  0.0190,  ...,  0.0069, -0.0068, -0.0118]]),\n",
       " 9598: tensor([[-0.0221,  0.0011,  0.0205,  ..., -0.0172, -0.0024, -0.0234]]),\n",
       " 317: tensor([[-0.0074,  0.0060,  0.0140,  ..., -0.0190, -0.0223, -0.0304]]),\n",
       " 7424: tensor([[-0.0245, -0.0070, -0.0040,  ..., -0.0118, -0.0115, -0.0066]]),\n",
       " 2621: tensor([[-0.0197,  0.0188,  0.0404,  ...,  0.0130, -0.0034, -0.0352]]),\n",
       " 9888: tensor([[-0.0120,  0.0003,  0.0371,  ..., -0.0260, -0.0301, -0.0188]]),\n",
       " 10886: tensor([[-0.0174,  0.0088,  0.0167,  ..., -0.0095, -0.0122, -0.0100]]),\n",
       " 7809: tensor([[-0.0210, -0.0088,  0.0342,  ..., -0.0163, -0.0141, -0.0423]]),\n",
       " 5385: tensor([[-0.0441, -0.0061,  0.0058,  ..., -0.0235, -0.0071, -0.0066]]),\n",
       " 4926: tensor([[-0.0077, -0.0186, -0.0028,  ..., -0.0131, -0.0045, -0.0170]]),\n",
       " 6153: tensor([[-0.0005, -0.0048,  0.0093,  ..., -0.0255, -0.0071, -0.0193]]),\n",
       " 3356: tensor([[-0.0127,  0.0059,  0.0100,  ..., -0.0333, -0.0147, -0.0125]]),\n",
       " 10752: tensor([[-0.0374,  0.0016, -0.0066,  ..., -0.0121, -0.0248, -0.0154]]),\n",
       " 348: tensor([[ 0.0057, -0.0226,  0.0042,  ..., -0.0151, -0.0046, -0.0436]]),\n",
       " 2897: tensor([[ 0.0009,  0.0078,  0.0330,  ..., -0.0305,  0.0164,  0.0060]]),\n",
       " 10630: tensor([[-0.0288,  0.0033,  0.0089,  ..., -0.0344, -0.0089, -0.0120]]),\n",
       " 3234: tensor([[-0.0188,  0.0075,  0.0169,  ..., -0.0014, -0.0086, -0.0108]]),\n",
       " 6031: tensor([[-0.0016, -0.0053,  0.0342,  ..., -0.0242, -0.0019, -0.0089]]),\n",
       " 4844: tensor([[-0.0061, -0.0047,  0.0014,  ...,  0.0070, -0.0089, -0.0284]]),\n",
       " 2743: tensor([[ 0.0167, -0.0267,  0.0263,  ..., -0.0033, -0.0173, -0.0261]]),\n",
       " 7546: tensor([[-0.0138,  0.0090,  0.0233,  ..., -0.0180, -0.0101, -0.0235]]),\n",
       " 4790: tensor([[-0.0009,  0.0148,  0.0297,  ..., -0.0179, -0.0278, -0.0368]]),\n",
       " 1595: tensor([[-0.0080, -0.0191, -0.0037,  ...,  0.0114, -0.0047, -0.0326]]),\n",
       " 4469: tensor([[ 0.0040, -0.0127,  0.0145,  ..., -0.0295,  0.0014,  0.0016]]),\n",
       " 8349: tensor([[-0.0226, -0.0185,  0.0229,  ..., -0.0078, -0.0235, -0.0427]]),\n",
       " 1789: tensor([[ 0.0109, -0.0208,  0.0334,  ..., -0.0155, -0.0247, -0.0244]]),\n",
       " 8155: tensor([[-0.0379, -0.0027,  0.0277,  ..., -0.0138,  0.0029, -0.0259]]),\n",
       " 4675: tensor([[-0.0001,  0.0010,  0.0184,  ..., -0.0216,  0.0050, -0.0226]]),\n",
       " 10801: tensor([[-0.0165, -0.0043,  0.0189,  ..., -0.0093, -0.0210, -0.0248]]),\n",
       " 1470: tensor([[-0.0391,  0.0062,  0.0164,  ..., -0.0107,  0.0121, -0.0070]]),\n",
       " 9422: tensor([[-0.0304, -0.0050,  0.0099,  ..., -0.0195,  0.0107, -0.0427]]),\n",
       " 5302: tensor([[-0.0024,  0.0084,  0.0376,  ..., -0.0109, -0.0365, -0.0251]]),\n",
       " 154: tensor([[-0.0491, -0.0323,  0.0154,  ...,  0.0004,  0.0078, -0.0296]]),\n",
       " 2972: tensor([[ 0.0175, -0.0080,  0.0518,  ...,  0.0063, -0.0112, -0.0348]]),\n",
       " 3028: tensor([[ 0.0103,  0.0009,  0.0356,  ..., -0.0173, -0.0131, -0.0392]]),\n",
       " 5199: tensor([[-0.0044, -0.0084,  0.0343,  ..., -0.0262, -0.0123, -0.0248]]),\n",
       " 2810: tensor([[-0.0180,  0.0006,  0.0057,  ...,  0.0012,  0.0014, -0.0377]]),\n",
       " 9540: tensor([[-0.0374,  0.0189, -0.0020,  ..., -0.0127, -0.0022, -0.0190]]),\n",
       " 5260: tensor([[-0.0240, -0.0153,  0.0212,  ...,  0.0054, -0.0108, -0.0235]]),\n",
       " 11039: tensor([[-0.0255,  0.0090,  0.0070,  ..., -0.0185,  0.0273, -0.0002]]),\n",
       " 7638: tensor([[-0.0264, -0.0116,  0.0196,  ..., -0.0340, -0.0216, -0.0162]]),\n",
       " 10963: tensor([[-0.0155,  0.0058,  0.0213,  ..., -0.0095, -0.0168, -0.0290]]),\n",
       " 1512: tensor([[ 0.0087,  0.0091,  0.0013,  ..., -0.0407, -0.0217,  0.0062]]),\n",
       " 8037: tensor([[-0.0288, -0.0015,  0.0203,  ..., -0.0187, -0.0236, -0.0391]]),\n",
       " 4717: tensor([[-0.0065,  0.0070,  0.0175,  ..., -0.0262,  0.0026, -0.0119]]),\n",
       " 3077: tensor([[ 0.0052,  0.0260,  0.0245,  ...,  0.0024, -0.0120, -0.0533]]),\n",
       " 1802: tensor([[-0.0189,  0.0007,  0.0177,  ..., -0.0053, -0.0023, -0.0370]]),\n",
       " 10473: tensor([[-0.0074, -0.0186,  0.0177,  ...,  0.0030, -0.0161, -0.0315]]),\n",
       " 6272: tensor([[-0.0257, -0.0126,  0.0032,  ..., -0.0250, -0.0319, -0.0106]]),\n",
       " 9784: tensor([[-0.0270,  0.0305,  0.0098,  ..., -0.0175, -0.0023, -0.0184]]),\n",
       " 2500: tensor([[-0.0031, -0.0081,  0.0351,  ..., -0.0284, -0.0141, -0.0328]]),\n",
       " 11104: tensor([[-0.0382,  0.0058,  0.0184,  ..., -0.0298, -0.0040, -0.0228]]),\n",
       " 9850: tensor([[-0.0197, -0.0178,  0.0047,  ..., -0.0273, -0.0078, -0.0016]]),\n",
       " 7705: tensor([[-0.0309, -0.0126,  0.0223,  ..., -0.0125, -0.0155, -0.0371]]),\n",
       " 4748: tensor([[-0.0067, -0.0134,  0.0375,  ..., -0.0100, -0.0149, -0.0010]]),\n",
       " 8068: tensor([[-0.0310, -0.0168,  0.0055,  ..., -0.0231, -0.0265, -0.0040]]),\n",
       " 8391: tensor([[-0.0004,  0.0106,  0.0230,  ..., -0.0096, -0.0053,  0.0018]]),\n",
       " 7667: tensor([[-0.0128, -0.0282,  0.0335,  ..., -0.0168, -0.0031, -0.0166]]),\n",
       " 9932: tensor([[-0.0374,  0.0095,  0.0226,  ..., -0.0164, -0.0083, -0.0176]]),\n",
       " 11066: tensor([[-0.0022,  0.0020,  0.0419,  ..., -0.0212, -0.0180, -0.0097]]),\n",
       " 2462: tensor([[-0.0105, -0.0044, -0.0078,  ...,  0.0004, -0.0272, -0.0072]]),\n",
       " 6310: tensor([[-0.0119, -0.0168,  0.0172,  ..., -0.0186, -0.0119, -0.0203]]),\n",
       " 1960: tensor([[ 0.0089, -0.0142, -0.0037,  ..., -0.0281, -0.0172, -0.0416]]),\n",
       " 10511: tensor([[-0.0358, -0.0147,  0.0283,  ..., -0.0240, -0.0171, -0.0275]]),\n",
       " 3115: tensor([[-0.0104, -0.0203,  0.0059,  ..., -0.0250, -0.0008, -0.0258]]),\n",
       " 390: tensor([[-0.0198,  0.0278,  0.0181,  ..., -0.0030, -0.0203, -0.0276]]),\n",
       " 5053: tensor([[-0.0198,  0.0114,  0.0105,  ..., -0.0261, -0.0111, -0.0181]]),\n",
       " 9773: tensor([[-0.0313, -0.0109,  0.0027,  ..., -0.0395, -0.0197, -0.0078]]),\n",
       " 7826: tensor([[-0.0280, -0.0238,  0.0004,  ..., -0.0135, -0.0163, -0.0101]]),\n",
       " 205: tensor([[-0.0225,  0.0230,  0.0212,  ..., -0.0212, -0.0060, -0.0443]]),\n",
       " 3080: tensor([[-0.0146,  0.0046,  0.0323,  ..., -0.0132, -0.0266, -0.0205]]),\n",
       " 4909: tensor([[-0.0014,  0.0020,  0.0203,  ..., -0.0164, -0.0192, -0.0333]]),\n",
       " 10484: tensor([[-0.0221,  0.0052,  0.0258,  ..., -0.0215, -0.0107, -0.0271]]),\n",
       " 6285: tensor([[-0.0263, -0.0122,  0.0146,  ..., -0.0168, -0.0238, -0.0151]]),\n",
       " 3379: tensor([[-0.0205,  0.0043,  0.0242,  ..., -0.0218, -0.0111, -0.0203]]),\n",
       " 4524: tensor([[-0.0385, -0.0049, -0.0024,  ..., -0.0177, -0.0061, -0.0161]]),\n",
       " 8204: tensor([[ 0.0033, -0.0020,  0.0258,  ..., -0.0047, -0.0166, -0.0307]]),\n",
       " 1721: tensor([[ 0.0075,  0.0008,  0.0297,  ...,  0.0110, -0.0025, -0.0160]]),\n",
       " 7690: tensor([[-0.0219,  0.0118,  0.0299,  ..., -0.0167, -0.0215, -0.0433]]),\n",
       " 11091: tensor([[-0.0323, -0.0082,  0.0054,  ..., -0.0372,  0.0109, -0.0163]]),\n",
       " 7569: tensor([[-0.0220, -0.0050,  0.0155,  ..., -0.0306, -0.0123, -0.0078]]),\n",
       " 2495: tensor([[-0.0330, -0.0093,  0.0109,  ..., -0.0210, -0.0351,  0.0038]]),\n",
       " 1643: tensor([[-0.0132, -0.0114,  0.0310,  ..., -0.0413, -0.0025, -0.0026]]),\n",
       " 52: tensor([[-0.0260,  0.0103,  0.0397,  ..., -0.0108, -0.0161, -0.0299]]),\n",
       " 4446: tensor([[-0.0185, -0.0243,  0.0087,  ..., -0.0442, -0.0111, -0.0154]]),\n",
       " 8366: tensor([[-0.0221,  0.0027,  0.0274,  ..., -0.0053, -0.0069, -0.0239]]),\n",
       " 367: tensor([[-0.0121, -0.0268,  0.0224,  ..., -0.0175, -0.0059, -0.0072]]),\n",
       " 7944: tensor([[-0.0387,  0.0002,  0.0160,  ..., -0.0125, -0.0143, -0.0226]]),\n",
       " 5131: tensor([[-0.0177, -0.0172,  0.0152,  ..., -0.0282, -0.0068, -0.0056]]),\n",
       " 9611: tensor([[-0.0165,  0.0055,  0.0354,  ..., -0.0092, -0.0105, -0.0212]]),\n",
       " 1997: tensor([[-0.0294, -0.0044,  0.0027,  ..., -0.0156, -0.0177, -0.0176]]),\n",
       " 4682: tensor([[-0.0087, -0.0132,  0.0216,  ..., -0.0096, -0.0194, -0.0093]]),\n",
       " 1487: tensor([[ 0.0138,  0.0009,  0.0337,  ..., -0.0119, -0.0153,  0.0008]]),\n",
       " 2651: tensor([[-0.0329, -0.0021, -0.0073,  ..., -0.0135,  0.0073, -0.0263]]),\n",
       " 7454: tensor([[ 0.0118,  0.0091,  0.0459,  ..., -0.0200, -0.0183, -0.0125]]),\n",
       " 3326: tensor([[ 0.0028, -0.0030,  0.0301,  ..., -0.0027, -0.0189, -0.0270]]),\n",
       " 10722: tensor([[-0.0231,  0.0073,  0.0181,  ..., -0.0098, -0.0019,  0.0058]]),\n",
       " 4956: tensor([[-0.0276, -0.0032,  0.0219,  ...,  0.0002, -0.0052, -0.0268]]),\n",
       " 6123: tensor([[-0.0275,  0.0033,  0.0251,  ..., -0.0218, -0.0217, -0.0311]]),\n",
       " 2985: tensor([[-0.0116, -0.0082,  0.0063,  ..., -0.0173, -0.0136, -0.0081]]),\n",
       " 7879: tensor([[-0.0007,  0.0156,  0.0003,  ..., -0.0157, -0.0017, -0.0269]]),\n",
       " 6041: tensor([[ 0.0030,  0.0171,  0.0338,  ..., -0.0031, -0.0208, -0.0437]]),\n",
       " 4834: tensor([[-0.0226, -0.0011, -0.0022,  ..., -0.0281, -0.0139, -0.0113]]),\n",
       " 10640: tensor([[-0.0185,  0.0036,  0.0204,  ..., -0.0058, -0.0047, -0.0048]]),\n",
       " 3244: tensor([[-0.0324, -0.0141,  0.0140,  ..., -0.0221, -0.0110, -0.0214]]),\n",
       " 338: tensor([[ 0.0081, -0.0071,  0.0269,  ..., -0.0254, -0.0137, -0.0254]]),\n",
       " 5297: tensor([[-0.0080, -0.0006,  0.0144,  ..., -0.0158, -0.0208, -0.0223]]),\n",
       " 10994: tensor([[-0.0354, -0.0110,  0.0072,  ..., -0.0258, -0.0328, -0.0233]]),\n",
       " 8339: tensor([[ 0.0058, -0.0109,  0.0196,  ...,  0.0024, -0.0157, -0.0082]]),\n",
       " 4419: tensor([[-0.0300,  0.0005,  0.0058,  ..., -0.0050, -0.0133, -0.0175]]),\n",
       " 7536: tensor([[-0.0354,  0.0092,  0.0270,  ..., -0.0261, -0.0119, -0.0335]]),\n",
       " 2733: tensor([[ 0.0041,  0.0159,  0.0413,  ..., -0.0125, -0.0116, -0.0223]]),\n",
       " 694: tensor([[ 0.0033,  0.0048,  0.0303,  ...,  0.0109, -0.0194, -0.0281]]),\n",
       " 8941: tensor([[-0.0106, -0.0128,  0.0237,  ..., -0.0329, -0.0091, -0.0183]]),\n",
       " 6614: tensor([[-0.0063, -0.0333,  0.0331,  ..., -0.0005, -0.0148, -0.0290]]),\n",
       " 3411: tensor([[ 0.0065,  0.0124,  0.0127,  ..., -0.0326, -0.0206, -0.0138]]),\n",
       " 10015: tensor([[-0.0442,  0.0030, -0.0119,  ..., -0.0085, -0.0165, -0.0178]]),\n",
       " 7363: tensor([[-0.0067,  0.0119,  0.0417,  ..., -0.0285, -0.0221, -0.0317]]),\n",
       " 940: tensor([[-0.0240,  0.0043,  0.0123,  ..., -0.0326, -0.0069, -0.0242]]),\n",
       " 2166: tensor([[-0.0139,  0.0015,  0.0044,  ..., -0.0076, -0.0135, -0.0091]]),\n",
       " 8695: tensor([[-0.0294,  0.0124,  0.0033,  ..., -0.0303, -0.0160, -0.0265]]),\n",
       " 1049: tensor([[-0.0222, -0.0006,  0.0116,  ..., -0.0275, -0.0052, -0.0349]]),\n",
       " 2004: tensor([[ 0.0087,  0.0176,  0.0277,  ..., -0.0270, -0.0204, -0.0326]]),\n",
       " 822: tensor([[-0.0562, -0.0082, -0.0044,  ..., -0.0387,  0.0143, -0.0094]]),\n",
       " 7201: tensor([[-0.0015, -0.0181,  0.0250,  ..., -0.0214, -0.0245, -0.0165]]),\n",
       " 9280: tensor([[-0.0245,  0.0190,  0.0223,  ...,  0.0062, -0.0100, -0.0369]]),\n",
       " 5659: tensor([[-0.0046,  0.0030,  0.0439,  ...,  0.0099, -0.0100, -0.0154]]),\n",
       " 9179: tensor([[-0.0243,  0.0183,  0.0235,  ..., -0.0036, -0.0243, -0.0133]]),\n",
       " 10177: tensor([[ 0.0055,  0.0026,  0.0170,  ..., -0.0043, -0.0104, -0.0207]]),\n",
       " 3573: tensor([[ 0.0006,  0.0016,  0.0279,  ..., -0.0107, -0.0169, -0.0099]]),\n",
       " 6776: tensor([[-0.0533,  0.0166, -0.0223,  ..., -0.0204, -0.0088, -0.0237]]),\n",
       " 8823: tensor([[-0.0215,  0.0085,  0.0220,  ..., -0.0170, -0.0193, -0.0301]]),\n",
       " 3863: tensor([[-0.0057, -0.0231,  0.0013,  ..., -0.0194, -0.0089, -0.0150]]),\n",
       " 1016: tensor([[ 0.0034, -0.0032,  0.0030,  ..., -0.0125, -0.0041, -0.0226]]),\n",
       " 8533: tensor([[-0.0154,  0.0009,  0.0261,  ..., -0.0375, -0.0097, -0.0193]]),\n",
       " 4213: tensor([[-0.0403, -0.0031, -0.0145,  ..., -0.0164,  0.0072, -0.0192]]),\n",
       " 2139: tensor([[-0.0329, -0.0022,  0.0283,  ..., -0.0178, -0.0206, -0.0040]]),\n",
       " 532: tensor([[ 0.0057,  0.0014,  0.0259,  ..., -0.0270, -0.0287, -0.0081]]),\n",
       " 9044: tensor([[-0.0228, -0.0061,  0.0114,  ..., -0.0230, -0.0009, -0.0061]]),\n",
       " 5764: tensor([[-0.0185, -0.0026,  0.0206,  ..., -0.0136, -0.0164, -0.0030]]),\n",
       " 6729: tensor([[ 0.0185, -0.0181,  0.0216,  ..., -0.0225, -0.0200, -0.0298]]),\n",
       " 10128: tensor([[-0.0316,  0.0004,  0.0141,  ..., -0.0215, -0.0364, -0.0094]]),\n",
       " 9126: tensor([[-0.0058,  0.0166,  0.0299,  ...,  0.0079, -0.0295, -0.0148]]),\n",
       " 5606: tensor([[-0.0053, -0.0192,  0.0250,  ..., -0.0213, -0.0161, -0.0307]]),\n",
       " 450: tensor([[ 0.0125, -0.0407,  0.0461,  ..., -0.0005, -0.0051, -0.0200]]),\n",
       " 8451: tensor([[-0.0082, -0.0157,  0.0194,  ..., -0.0039, -0.0295, -0.0426]]),\n",
       " 4371: tensor([[-0.0147,  0.0054,  0.0122,  ..., -0.0318, -0.0113, -0.0292]]),\n",
       " 1174: tensor([[-0.0045, -0.0077,  0.0139,  ..., -0.0142, -0.0031, -0.0420]]),\n",
       " 4088: tensor([[ 0.0063,  0.0056,  0.0431,  ..., -0.0119, -0.0012, -0.0217]]),\n",
       " 3901: tensor([[-0.0280, -0.0144,  0.0213,  ..., -0.0246, -0.0152, -0.0310]]),\n",
       " 5847: tensor([[ 0.0061, -0.0010,  0.0194,  ..., -0.0158, -0.0033, -0.0202]]),\n",
       " 7032: tensor([[-0.0148, -0.0018,  0.0213,  ..., -0.0294, -0.0194, -0.0084]]),\n",
       " 2237: tensor([[-0.0350, -0.0165, -0.0155,  ...,  0.0048, -0.0082,  0.0012]]),\n",
       " 3894: tensor([[ 0.0117,  0.0188,  0.0387,  ..., -0.0024, -0.0187, -0.0320]]),\n",
       " 6968: tensor([[-0.0295,  0.0015,  0.0022,  ..., -0.0125,  0.0013, -0.0248]]),\n",
       " 1318: tensor([[-0.0478,  0.0027, -0.0109,  ..., -0.0156, -0.0205, -0.0156]]),\n",
       " 5793: tensor([[-0.0122, -0.0028,  0.0019,  ..., -0.0100, -0.0219, -0.0075]]),\n",
       " 6545: tensor([[ 0.0193, -0.0130,  0.0268,  ..., -0.0316,  0.0008, -0.0180]]),\n",
       " 3740: tensor([[-0.0242, -0.0092,  0.0377,  ...,  0.0072, -0.0067, -0.0265]]),\n",
       " 10344: tensor([[-0.0353,  0.0052,  0.0095,  ...,  0.0064, -0.0014, -0.0088]]),\n",
       " 9228: tensor([[-0.0328, -0.0024, -0.0050,  ..., -0.0120, -0.0046, -0.0154]]),\n",
       " 5508: tensor([[-0.0299,  0.0138,  0.0058,  ...,  0.0041, -0.0210, -0.0071]]),\n",
       " 10226: tensor([[ 0.0075,  0.0224,  0.0486,  ..., -0.0260, -0.0187, -0.0281]]),\n",
       " 3622: tensor([[-0.0009, -0.0045,  0.0284,  ..., -0.0026, -0.0180, -0.0265]]),\n",
       " 6427: tensor([[-0.0029, -0.0063,  0.0118,  ...,  0.0083, -0.0066, -0.0118]]),\n",
       " 2355: tensor([[ 0.0076,  0.0348,  0.0016,  ..., -0.0335, -0.0152, -0.0410]]),\n",
       " 7150: tensor([[-0.0409, -0.0054,  0.0122,  ...,  0.0009,  0.0034, -0.0136]]),\n",
       " 5925: tensor([[-0.0332,  0.0140,  0.0288,  ..., -0.0189, -0.0233, -0.0200]]),\n",
       " 4386: tensor([[-0.0399,  0.0036, -0.0054,  ...,  0.0006, -0.0149, -0.0200]]),\n",
       " 1183: tensor([[-0.0160, -0.0037,  0.0159,  ..., -0.0207, -0.0102, -0.0115]]),\n",
       " 663: tensor([[-0.0286,  0.0058, -0.0058,  ...,  0.0042, -0.0079, -0.0184]]),\n",
       " 5435: tensor([[-0.0157,  0.0067,  0.0173,  ..., -0.0217, -0.0171, -0.0020]]),\n",
       " 9315: tensor([[-0.0221, -0.0107, -0.0050,  ..., -0.0286, -0.0241, -0.0156]]),\n",
       " 1347: tensor([[ 0.0003,  0.0139,  0.0448,  ..., -0.0204, -0.0336, -0.0116]]),\n",
       " 4142: tensor([[-0.0460, -0.0034,  0.0008,  ..., -0.0081, -0.0197, -0.0351]]),\n",
       " 8662: tensor([[-0.0418,  0.0038, -0.0149,  ..., -0.0295, -0.0023, -0.0209]]),\n",
       " 6937: tensor([[ 0.0036,  0.0025,  0.0249,  ...,  0.0044, -0.0202, -0.0255]]),\n",
       " 7394: tensor([[-0.0136, -0.0024,  0.0071,  ..., -0.0067,  0.0006, -0.0367]]),\n",
       " 2268: tensor([[ 0.0031, -0.0061,  0.0343,  ..., -0.0076, -0.0148, -0.0226]]),\n",
       " 2191: tensor([[-0.0390, -0.0204,  0.0057,  ..., -0.0158, -0.0220, -0.0245]]),\n",
       " 5818: tensor([[-0.0170,  0.0178,  0.0281,  ..., -0.0163, -0.0281, -0.0152]]),\n",
       " 6855: tensor([[-0.0011,  0.0207,  0.0109,  ...,  0.0100, -0.0108, -0.0268]]),\n",
       " 4020: tensor([[-0.0179, -0.0182,  0.0189,  ..., -0.0200, -0.0175, -0.0196]]),\n",
       " 8700: tensor([[-0.0217,  0.0091,  0.0088,  ..., -0.0163, -0.0070, -0.0336]]),\n",
       " 1225: tensor([[-0.0184,  0.0183,  0.0108,  ..., -0.0092, -0.0136, -0.0214]]),\n",
       " 10180: tensor([[-0.0240,  0.0012,  0.0208,  ..., -0.0064, -0.0158, -0.0176]]),\n",
       " 6478: tensor([[-0.0193,  0.0060,  0.0203,  ..., -0.0034, -0.0302, -0.0210]]),\n",
       " 3584: tensor([[ 0.0087, -0.0020,  0.0377,  ..., -0.0201, -0.0118, -0.0126]]),\n",
       " 6781: tensor([[-0.0274,  0.0075,  0.0133,  ..., -0.0247, -0.0041, -0.0267]]),\n",
       " 10279: tensor([[-0.0089,  0.0089,  0.0200,  ..., -0.0165, -0.0006, -0.0309]]),\n",
       " 5557: tensor([[ 0.0015,  0.0105,  0.0346,  ..., -0.0343,  0.0032, -0.0232]]),\n",
       " 9277: tensor([[-0.0251,  0.0079,  0.0117,  ..., -0.0103, -0.0138, -0.0377]]),\n",
       " 701: tensor([[ 0.0023,  0.0103,  0.0042,  ..., -0.0120, -0.0272, -0.0222]]),\n",
       " 10735: tensor([[-0.0161, -0.0210,  0.0208,  ..., -0.0208, -0.0138, -0.0297]]),\n",
       " 3331: tensor([[-0.0189,  0.0013,  0.0055,  ..., -0.0188, -0.0082, -0.0030]]),\n",
       " 6134: tensor([[-0.0009, -0.0017,  0.0158,  ..., -0.0325, -0.0097, -0.0113]]),\n",
       " 4941: tensor([[-0.0059, -0.0107,  0.0391,  ..., -0.0285, -0.0184, -0.0323]]),\n",
       " 2992: tensor([[-0.0161,  0.0064,  0.0126,  ..., -0.0580, -0.0046,  0.0127]]),\n",
       " 4695: tensor([[ 0.0042,  0.0118,  0.0354,  ..., -0.0251,  0.0003, -0.0374]]),\n",
       " 1769: tensor([[ 0.0036, -0.0014,  0.0243,  ..., -0.0189, -0.0011, -0.0095]]),\n",
       " 1490: tensor([[ 0.0027,  0.0032,  0.0366,  ..., -0.0277, -0.0116, -0.0012]]),\n",
       " 2646: tensor([[-0.0048, -0.0232, -0.0155,  ..., -0.0172, -0.0092, -0.0123]]),\n",
       " 7443: tensor([[-0.0111,  0.0037,  0.0047,  ..., -0.0094,  0.0007, -0.0127]]),\n",
       " 10983: tensor([[-0.0136,  0.0026,  0.0018,  ...,  0.0003,  0.0057, -0.0099]]),\n",
       " 7521: tensor([[-0.0393, -0.0110,  0.0032,  ..., -0.0296, -0.0303, -0.0224]]),\n",
       " 2724: tensor([[-0.0145,  0.0066,  0.0112,  ..., -0.0145, -0.0203, -0.0112]]),\n",
       " 4823: tensor([[-0.0304, -0.0206,  0.0102,  ..., -0.0095, -0.0021, -0.0157]]),\n",
       " 6056: tensor([[-0.0437,  0.0114,  0.0105,  ..., -0.0220, -0.0154, -0.0119]]),\n",
       " 3253: tensor([[-0.0042, -0.0083,  0.0156,  ..., -0.0398, -0.0189, -0.0195]]),\n",
       " 10657: tensor([[-0.0310, -0.0094, -0.0100,  ..., -0.0234,  0.0088, -0.0276]]),\n",
       " 5179: tensor([[ 0.0055,  0.0203,  0.0196,  ..., -0.0210, -0.0179, -0.0234]]),\n",
       " 9659: tensor([[-0.0324,  0.0023,  0.0240,  ..., -0.0297, -0.0088, -0.0302]]),\n",
       " 5280: tensor([[ 0.0022,  0.0022,  0.0212,  ..., -0.0279, -0.0041, -0.0182]]),\n",
       " 2619: tensor([[-0.0008,  0.0142, -0.0085,  ..., -0.0037, -0.0200, -0.0260]]),\n",
       " 8213: tensor([[-0.0089,  0.0283,  0.0268,  ..., -0.0264, -0.0060, -0.0423]]),\n",
       " 4533: tensor([[ 0.0051, -0.0012,  0.0039,  ..., -0.0186, -0.0019, -0.0091]]),\n",
       " 1736: tensor([[-0.0105, -0.0052,  0.0113,  ..., -0.0129, -0.0007, -0.0127]]),\n",
       " 7831: tensor([[-0.0204,  0.0090,  0.0237,  ..., -0.0178, -0.0194, -0.0192]]),\n",
       " 9764: tensor([[-0.0091,  0.0251,  0.0162,  ..., -0.0145, -0.0049, -0.0173]]),\n",
       " 5044: tensor([[-0.0006, -0.0005,  0.0270,  ..., -0.0270, -0.0206,  0.0006]]),\n",
       " 212: tensor([[-0.0527,  0.0063,  0.0105,  ..., -0.0196,  0.0023, -0.0156]]),\n",
       " 10493: tensor([[-0.0386,  0.0022,  0.0042,  ..., -0.0359, -0.0269, -0.0270]]),\n",
       " 3097: tensor([[-0.0284,  0.0068,  0.0119,  ..., -0.0427, -0.0388, -0.0188]]),\n",
       " 6292: tensor([[-0.0255, -0.0046,  0.0103,  ..., -0.0228, -0.0325, -0.0056]]),\n",
       " 370: tensor([[-0.0034, -0.0168,  0.0410,  ..., -0.0322, -0.0139, -0.0014]]),\n",
       " 9606: tensor([[-0.0086, -0.0089, -0.0003,  ..., -0.0184,  0.0055, -0.0173]]),\n",
       " 5126: tensor([[-0.0309, -0.0040,  0.0255,  ..., -0.0030,  0.0212, -0.0124]]),\n",
       " 7953: tensor([[-0.0240,  0.0061,  0.0039,  ..., -0.0102,  0.0039, -0.0012]]),\n",
       " 10608: tensor([[-0.0296, -0.0043,  0.0114,  ..., -0.0008,  0.0044, -0.0236]]),\n",
       " 6009: tensor([[-0.0071, -0.0012,  0.0204,  ..., -0.0217,  0.0035, -0.0319]]),\n",
       " 1980: tensor([[ 0.0078,  0.0113,  0.0308,  ..., -0.0318, -0.0032, -0.0260]]),\n",
       " 7687: tensor([[-0.0087,  0.0054,  0.0217,  ...,  0.0185, -0.0028, -0.0314]]),\n",
       " 2482: tensor([[-0.0387, -0.0023,  0.0166,  ..., -0.0144,  0.0267, -0.0210]]),\n",
       " 11086: tensor([[-0.0111, -0.0092,  0.0257,  ..., -0.0173, -0.0097, -0.0111]]),\n",
       " 1654: tensor([[-0.0120,  0.0126,  0.0308,  ...,  0.0039, -0.0085, -0.0489]]),\n",
       " 8088: tensor([[-0.0282,  0.0200,  0.0099,  ..., -0.0317, -0.0001, -0.0319]]),\n",
       " 45: tensor([[-0.0412,  0.0007,  0.0022,  ...,  0.0035, -0.0248, -0.0187]]),\n",
       " 8371: tensor([[-0.0335, -0.0050,  0.0104,  ..., -0.0267, -0.0038, -0.0287]]),\n",
       " 4451: tensor([[-0.0070,  0.0349,  0.0299,  ...,  0.0039, -0.0060, -0.0375]]),\n",
       " 1438: tensor([[ 0.0013,  0.0030,  0.0447,  ...,  0.0028,  0.0058, -0.0422]]),\n",
       " 10849: tensor([[-0.0316, -0.0024,  0.0365,  ..., -0.0080,  0.0076, -0.0165]]),\n",
       " 11113: tensor([[-0.0383, -0.0023,  0.0148,  ..., -0.0077, -0.0080, -0.0307]]),\n",
       " 2517: tensor([[-0.0022,  0.0098,  0.0129,  ..., -0.0083, -0.0102, -0.0309]]),\n",
       " 7712: tensor([[-0.0275, -0.0078,  0.0033,  ..., -0.0127, -0.0268, -0.0105]]),\n",
       " 9847: tensor([[ 0.0179,  0.0145,  0.0304,  ..., -0.0107, -0.0012, -0.0037]]),\n",
       " 1815: tensor([[-0.0157,  0.0050,  0.0150,  ..., -0.0129, -0.0170, -0.0090]]),\n",
       " 10464: tensor([[-0.0177,  0.0101,  0.0261,  ..., -0.0278, -0.0159, -0.0403]]),\n",
       " 3060: tensor([[-0.0073, -0.0203,  0.0246,  ..., -0.0055, -0.0149, -0.0291]]),\n",
       " 3399: tensor([[-0.0469, -0.0068, -0.0149,  ..., -0.0305, -0.0085,  0.0092]]),\n",
       " 6265: tensor([[-0.0275, -0.0191,  0.0355,  ..., -0.0001, -0.0013, -0.0057]]),\n",
       " 9793: tensor([[-0.0151, -0.0172,  0.0018,  ..., -0.0070, -0.0188, -0.0212]]),\n",
       " 6307: tensor([[-0.0127,  0.0069,  0.0152,  ..., -0.0273, -0.0079, -0.0445]]),\n",
       " 3102: tensor([[-0.0006, -0.0165,  0.0378,  ..., -0.0216, -0.0184, -0.0309]]),\n",
       " 1977: tensor([[-0.0361, -0.0019,  0.0303,  ..., -0.0246, -0.0069, -0.0199]]),\n",
       " 10506: tensor([[-0.0097, -0.0021,  0.0033,  ..., -0.0258,  0.0016, -0.0316]]),\n",
       " 387: tensor([[-0.0224, -0.0049,  0.0097,  ..., -0.0226, -0.0163, -0.0153]]),\n",
       " 9508: tensor([[-0.0109, -0.0113,  0.0182,  ..., -0.0082,  0.0021, -0.0163]]),\n",
       " 5228: tensor([[-0.0150, -0.0070,  0.0167,  ..., -0.0211, -0.0322, -0.0232]]),\n",
       " 2858: tensor([[-0.0048, -0.0030, -0.0138,  ..., -0.0304, -0.0283, -0.0043]]),\n",
       " 8386: tensor([[-0.0245,  0.0052,  0.0202,  ..., -0.0046, -0.0019, -0.0299]]),\n",
       " 9925: tensor([[-0.0152,  0.0098,  0.0410,  ..., -0.0241, -0.0094, -0.0388]]),\n",
       " 7670: tensor([[-0.0242, -0.0101, -0.0025,  ...,  0.0021, -0.0071, -0.0046]]),\n",
       " 2475: tensor([[ 0.0029,  0.0205,  0.0181,  ..., -0.0058, -0.0129, -0.0294]]),\n",
       " 7589: tensor([[-0.0176,  0.0111,  0.0183,  ..., -0.0037, -0.0142, -0.0091]]),\n",
       " 11071: tensor([[-0.0195,  0.0077,  0.0212,  ..., -0.0311, -0.0059, -0.0054]]),\n",
       " 5315: tensor([[ 0.0101, -0.0116,  0.0136,  ..., -0.0163, -0.0181, -0.0292]]),\n",
       " 9435: tensor([[-0.0182, -0.0162,  0.0272,  ..., -0.0138, -0.0126, -0.0279]]),\n",
       " 7899: tensor([[-0.0237, -0.0061,  0.0249,  ..., -0.0126, -0.0215, -0.0444]]),\n",
       " 2965: tensor([[-0.0387, -0.0125, -0.0120,  ..., -0.0103, -0.0049, -0.0148]]),\n",
       " 143: tensor([[-0.0033,  0.0105,  0.0088,  ..., -0.0159, -0.0337, -0.0148]]),\n",
       " 9818: tensor([[-0.0175,  0.0015,  0.0224,  ..., -0.0183, -0.0113, -0.0290]]),\n",
       " 2548: tensor([[-0.0351, -0.0132,  0.0128,  ...,  0.0036, -0.0071, -0.0065]]),\n",
       " 4662: tensor([[-0.0264, -0.0017, -0.0042,  ..., -0.0113, -0.0254, -0.0260]]),\n",
       " 8142: tensor([[-0.0105, -0.0028, -0.0144,  ..., -0.0323, -0.0280, -0.0084]]),\n",
       " 10816: tensor([[-0.0142,  0.0040, -0.0068,  ..., -0.0130,  0.0143, -0.0369]]),\n",
       " 1467: tensor([[ 7.3645e-05, -1.4166e-03,  4.0566e-02,  ..., -1.3547e-02,\n",
       "          -2.7496e-02, -1.0925e-02]]),\n",
       " 10974: tensor([[ 0.0086, -0.0010,  0.0288,  ..., -0.0174, -0.0090, -0.0194]]),\n",
       " 1505: tensor([[-0.0200, -0.0319, -0.0020,  ..., -0.0139, -0.0024, -0.0242]]),\n",
       " 4700: tensor([[-0.0158, -0.0016,  0.0392,  ..., -0.0258, -0.0106, -0.0043]]),\n",
       " 8020: tensor([[-0.0130,  0.0055, -0.0011,  ..., -0.0103,  0.0199, -0.0016]]),\n",
       " 2807: tensor([[-0.0361, -0.0139,  0.0135,  ..., -0.0176, -0.0162,  0.0044]]),\n",
       " 5277: tensor([[-0.0277, -0.0066, -0.0004,  ..., -0.0136, -0.0242,  0.0082]]),\n",
       " 9557: tensor([[ 0.0090,  0.0063,  0.0526,  ..., -0.0224, -0.0060, -0.0244]]),\n",
       " 10559: tensor([[-0.0158,  0.0038,  0.0182,  ..., -0.0307, -0.0057, -0.0037]]),\n",
       " 1928: tensor([[-0.0127, -0.0120,  0.0086,  ...,  0.0048, -0.0144, -0.0179]]),\n",
       " 6358: tensor([[-0.0204,  0.0154,  0.0100,  ..., -0.0009, -0.0142, -0.0165]]),\n",
       " 1350: tensor([[-0.0051,  0.0061,  0.0221,  ..., -0.0054, -0.0087, -0.0124]]),\n",
       " 6920: tensor([[-0.0147, -0.0006,  0.0008,  ..., -0.0139, -0.0115, -0.0318]]),\n",
       " 8675: tensor([[-0.0336, -0.0047, -0.0085,  ..., -0.0092, -0.0134, -0.0094]]),\n",
       " 4155: tensor([[-0.0127, -0.0147,  0.0198,  ..., -0.0041, -0.0022, -0.0198]]),\n",
       " 7383: tensor([[-0.0300,  0.0139,  0.0403,  ..., -0.0127, -0.0246, -0.0191]]),\n",
       " 2186: tensor([[ 0.0027, -0.0017,  0.0390,  ..., -0.0045, -0.0129, -0.0244]]),\n",
       " 3708: tensor([[-0.0254,  0.0006,  0.0179,  ..., -0.0231, -0.0207, -0.0087]]),\n",
       " 674: tensor([[-0.0179,  0.0102,  0.0211,  ..., -0.0282, -0.0186, -0.0182]]),\n",
       " 9302: tensor([[-0.0195, -0.0082,  0.0227,  ..., -0.0416, -0.0221, -0.0411]]),\n",
       " 5422: tensor([[-0.0122, -0.0093,  0.0146,  ...,  0.0143, -0.0182, -0.0193]]),\n",
       " 3593: tensor([[-0.0158, -0.0067,  0.0134,  ...,  0.0115, -0.0129, -0.0189]]),\n",
       " 10197: tensor([[-0.0295, -0.0013, -0.0104,  ..., -0.0173, -0.0173, -0.0186]]),\n",
       " 6796: tensor([[-0.0237,  0.0040,  0.0229,  ..., -0.0134, -0.0225, -0.0281]]),\n",
       " 9260: tensor([[ 0.0113, -0.0028,  0.0248,  ..., -0.0228, -0.0097, -0.0258]]),\n",
       " 5540: tensor([[ 0.0077,  0.0039,  0.0351,  ...,  0.0018, -0.0281, -0.0167]]),\n",
       " 716: tensor([[-0.0170,  0.0058,  0.0154,  ..., -0.0117, -0.0237, -0.0067]]),\n",
       " 9199: tensor([[-0.0344, -0.0068,  0.0163,  ..., -0.0327, -0.0194,  0.0024]]),\n",
       " 8717: tensor([[-0.0175, -0.0209,  0.0197,  ..., -0.0102, -0.0079, -0.0212]]),\n",
       " 4037: tensor([[-0.0025, -0.0031,  0.0100,  ..., -0.0090, -0.0219, -0.0005]]),\n",
       " 6842: tensor([[-0.0355,  0.0048,  0.0144,  ..., -0.0155, -0.0137, -0.0117]]),\n",
       " 1232: tensor([[ 0.0013, -0.0125,  0.0153,  ..., -0.0189, -0.0137, -0.0079]]),\n",
       " 7118: tensor([[-0.0232,  0.0083,  0.0139,  ..., -0.0156, -0.0284, -0.0190]]),\n",
       " 5784: tensor([[-0.0177, -0.0242,  0.0051,  ..., -0.0073, -0.0140, -0.0116]]),\n",
       " 6552: tensor([[-0.0281, -0.0099,  0.0041,  ..., -0.0153,  0.0009, -0.0312]]),\n",
       " 10353: tensor([[-0.0113,  0.0064,  0.0207,  ..., -0.0114, -0.0112, -0.0311]]),\n",
       " 3757: tensor([[ 0.0020,  0.0010,  0.0364,  ..., -0.0367, -0.0053, -0.0115]]),\n",
       " 7025: tensor([[-0.0112, -0.0213,  0.0312,  ..., -0.0231, -0.0169, -0.0055]]),\n",
       " 5850: tensor([[-0.0248,  0.0059,  0.0139,  ..., -0.0342, -0.0046, -0.0217]]),\n",
       " 2220: tensor([[ 0.0010,  0.0152,  0.0134,  ..., -0.0103, -0.0234, -0.0220]]),\n",
       " 3883: tensor([[-0.0183, -0.0010,  0.0268,  ..., -0.0044, -0.0150, -0.0548]]),\n",
       " 2342: tensor([[-0.0557,  0.0016, -0.0076,  ..., -0.0089, -0.0155, -0.0006]]),\n",
       " 5932: tensor([[-0.0336,  0.0026,  0.0260,  ..., -0.0391, -0.0105, -0.0179]]),\n",
       " 7147: tensor([[ 0.0135,  0.0089,  0.0252,  ..., -0.0082, -0.0091, -0.0266]]),\n",
       " 4391: tensor([[-0.0193, -0.0099,  0.0175,  ..., -0.0085, -0.0255, -0.0164]]),\n",
       " 4068: tensor([[-0.0382,  0.0068,  0.0127,  ..., -0.0167,  0.0033, -0.0421]]),\n",
       " 8748: tensor([[-0.0178,  0.0105,  0.0146,  ...,  0.0007, -0.0042, -0.0263]]),\n",
       " 1194: tensor([[ 0.0084,  0.0062,  0.0376,  ..., -0.0229, -0.0254, -0.0276]]),\n",
       " 749: tensor([[-0.0192,  0.0010,  0.0106,  ..., -0.0230, -0.0273, -0.0294]]),\n",
       " 3635: tensor([[-0.0131,  0.0009, -0.0141,  ..., -0.0320, -0.0252,  0.0007]]),\n",
       " 10231: tensor([[-0.0269, -0.0014,  0.0215,  ..., -0.0193,  0.0006, -0.0173]]),\n",
       " 6430: tensor([[ 0.0003, -0.0066,  0.0102,  ..., -0.0498, -0.0109, -0.0318]]),\n",
       " 3459: tensor([[-0.0196,  0.0142,  0.0241,  ...,  0.0042, -0.0126, -0.0310]]),\n",
       " 8909: tensor([[-0.0042, -0.0045,  0.0073,  ..., -0.0099, -0.0052, -0.0226]]),\n",
       " 525: tensor([[-0.0202, -0.0040,  0.0289,  ..., -0.0086, -0.0010, -0.0272]]),\n",
       " 5773: tensor([[-0.0372, -0.0025,  0.0106,  ..., -0.0217, -0.0219, -0.0091]]),\n",
       " 9053: tensor([[-0.0063,  0.0033,  0.0367,  ...,  0.0109, -0.0006, -0.0106]]),\n",
       " 1001: tensor([[ 0.0106, -0.0263,  0.0300,  ..., -0.0143, -0.0110, -0.0239]]),\n",
       " 6988: tensor([[-0.0085, -0.0150,  0.0354,  ..., -0.0231, -0.0136,  0.0107]]),\n",
       " 3874: tensor([[-0.0167,  0.0257,  0.0180,  ...,  0.0031, -0.0217, -0.0280]]),\n",
       " 4204: tensor([[-0.0272, -0.0008,  0.0129,  ..., -0.0057,  0.0006, -0.0305]]),\n",
       " 8524: tensor([[-0.0041, -0.0217,  0.0227,  ..., -0.0109, -0.0042, -0.0103]]),\n",
       " 908: tensor([[-0.0077, -0.0233,  0.0289,  ...,  0.0083, -0.0163, -0.0348]]),\n",
       " 4366: tensor([[-0.0108, -0.0213, -0.0179,  ..., -0.0097, -0.0104, -0.0228]]),\n",
       " 8446: tensor([[-0.0203,  0.0017,  0.0066,  ..., -0.0374, -0.0040, -0.0095]]),\n",
       " 3916: tensor([[-0.0292, -0.0062,  0.0114,  ..., -0.0065, -0.0101, -0.0142]]),\n",
       " 1163: tensor([[-0.0311, -0.0078,  0.0090,  ..., -0.0087, -0.0367, -0.0042]]),\n",
       " 7249: tensor([[-0.0363,  0.0007,  0.0210,  ..., -0.0258, -0.0082, -0.0190]]),\n",
       " 5611: tensor([[-0.0021, -0.0112,  0.0101,  ..., -0.0292, -0.0181, -0.0119]]),\n",
       " 9131: tensor([[ 0.0007, -0.0110,  0.0336,  ..., -0.0266,  0.0026, -0.0446]]),\n",
       " 447: tensor([[-0.0038,  0.0021,  0.0029,  ..., -0.0194, -0.0162, -0.0112]]),\n",
       " 2288: tensor([[ 0.0033,  0.0203,  0.0269,  ..., -0.0081,  0.0002, -0.0238]]),\n",
       " 7374: tensor([[-0.0269, -0.0135, -0.0030,  ..., -0.0432, -0.0121, -0.0440]]),\n",
       " 2171: tensor([[ 0.0010, -0.0059,  0.0329,  ..., -0.0062, -0.0254, -0.0355]]),\n",
       " 957: tensor([[-0.0215, -0.0118,  0.0199,  ..., -0.0138, -0.0028, -0.0091]]),\n",
       " 8682: tensor([[-0.0142,  0.0059,  0.0289,  ..., -0.0200, -0.0324, -0.0149]]),\n",
       " 683: tensor([[-0.0431, -0.0086,  0.0031,  ..., -0.0066,  0.0002, -0.0061]]),\n",
       " 6603: tensor([[-0.0238, -0.0063,  0.0207,  ..., -0.0050, -0.0266, -0.0393]]),\n",
       " 8956: tensor([[-0.0071, -0.0141,  0.0208,  ..., -0.0214,  0.0026, -0.0357]]),\n",
       " 10002: tensor([[-0.0179, -0.0082,  0.0165,  ..., -0.0229,  0.0065, -0.0336]]),\n",
       " 3406: tensor([[ 0.0017, -0.0031,  0.0586,  ..., -0.0147, -0.0047,  0.0002]]),\n",
       " 9297: tensor([[ 0.0071,  0.0061,  0.0431,  ..., -0.0067, -0.0237, -0.0161]]),\n",
       " 418: tensor([[-0.0220, -0.0229,  0.0110,  ..., -0.0161, -0.0192, -0.0204]]),\n",
       " 3564: tensor([[-0.0166, -0.0146, -0.0007,  ..., -0.0112, -0.0097, -0.0310]]),\n",
       " 6498: tensor([[-0.0192,  0.0027,  0.0286,  ..., -0.0177, -0.0002, -0.0312]]),\n",
       " 10160: tensor([[-3.6370e-02,  1.7395e-02,  3.0723e-02,  ..., -9.7594e-03,\n",
       "           7.5133e-05, -3.6059e-02]]),\n",
       " 8834: tensor([[-0.0427,  0.0027,  0.0098,  ..., -0.0112, -0.0154, -0.0249]]),\n",
       " 10299: tensor([[-0.0330,  0.0009,  0.0054,  ..., -0.0074, -0.0110, -0.0285]]),\n",
       " 6761: tensor([[-0.0170, -0.0088,  0.0253,  ..., -0.0199, -0.0046, -0.0139]]),\n",
       " 835: tensor([[-0.0255, -0.0135,  0.0244,  ..., -0.0026, -0.0110, -0.0170]]),\n",
       " 2013: tensor([[-0.0312, -0.0043,  0.0019,  ..., -0.0096, -0.0211, -0.0157]]),\n",
       " 7216: tensor([[-0.0109, -0.0108,  0.0227,  ..., -0.0082, -0.0135, -0.0203]]),\n",
       " 3949: tensor([[-0.0210, -0.0109, -0.0046,  ..., -0.0061, -0.0193, -0.0076]]),\n",
       " 8419: tensor([[-0.0302, -0.0029,  0.0033,  ...,  0.0026,  0.0173, -0.0254]]),\n",
       " 4339: tensor([[-0.0245,  0.0005, -0.0080,  ..., -0.0236, -0.0007, -0.0199]]),\n",
       " 6950: tensor([[-0.0178,  0.0049,  0.0232,  ..., -0.0036, -0.0318, -0.0300]]),\n",
       " 4125: tensor([[-0.0305,  0.0269,  0.0091,  ..., -0.0237, -0.0106, -0.0365]]),\n",
       " 8605: tensor([[ 0.0161,  0.0237,  0.0209,  ..., -0.0180, -0.0117, -0.0259]]),\n",
       " 1320: tensor([[-0.0296,  0.0143,  0.0070,  ..., -0.0468, -0.0141, -0.0256]]),\n",
       " 5452: tensor([[-0.0208, -0.0039,  0.0107,  ..., -0.0079, -0.0204, -0.0195]]),\n",
       " 9372: tensor([[-0.0035, -0.0138,  0.0279,  ..., -0.0024, -0.0228, -0.0198]]),\n",
       " 604: tensor([[ 0.0020,  0.0077,  0.0155,  ..., -0.0064, -0.0150, -0.0430]]),\n",
       " 10085: tensor([[-0.0370,  0.0243,  0.0034,  ..., -0.0140, -0.0142, -0.0190]]),\n",
       " 3481: tensor([[-0.0184, -0.0374,  0.0271,  ...,  0.0138,  0.0016, -0.0415]]),\n",
       " 3778: tensor([[-0.0197, -0.0067,  0.0051,  ..., -0.0292, -0.0158,  0.0036]]),\n",
       " 6684: tensor([[-0.0412,  0.0016,  0.0173,  ..., -0.0456, -0.0274,  0.0039]]),\n",
       " 766: tensor([[-0.0099,  0.0199,  0.0281,  ..., -0.0083, -0.0179, -0.0262]]),\n",
       " 5530: tensor([[ 0.0077,  0.0293,  0.0225,  ..., -0.0117, -0.0015,  0.0009]]),\n",
       " 9210: tensor([[-0.0152, -0.0021,  0.0331,  ..., -0.0085, -0.0169, -0.0051]]),\n",
       " 7291: tensor([[ 0.0194, -0.0191,  0.0522,  ..., -0.0241, -0.0045, -0.0031]]),\n",
       " 2094: tensor([[-0.0089, -0.0006,  0.0156,  ..., -0.0152,  0.0006, -0.0249]]),\n",
       " 7168: tensor([[-0.0105,  0.0050,  0.0332,  ...,  0.0005, -0.0131, -0.0195]]),\n",
       " 1242: tensor([[-0.0347, -0.0071,  0.0282,  ..., -0.0166, -0.0132, -0.0221]]),\n",
       " 4047: tensor([[-0.0303, -0.0152, -0.0130,  ..., -0.0188,  0.0007, -0.0148]]),\n",
       " 8767: tensor([[-0.0368,  0.0015,  0.0129,  ...,  0.0027,  0.0002, -0.0173]]),\n",
       " 6832: tensor([[-0.0263, -0.0101, -0.0025,  ..., -0.0193,  0.0158, -0.0184]]),\n",
       " 10323: tensor([[-0.0305, -0.0144, -0.0040,  ..., -0.0310, -0.0024, -0.0280]]),\n",
       " 3727: tensor([[ 0.0009,  0.0180,  0.0425,  ...,  0.0013,  0.0099, -0.0379]]),\n",
       " 6522: tensor([[-0.0439,  0.0192,  0.0211,  ...,  0.0002, -0.0213, -0.0177]]),\n",
       " 4283: tensor([[ 0.0131, -0.0037,  0.0229,  ..., -0.0024, -0.0084, -0.0380]]),\n",
       " 1086: tensor([[-0.0041,  0.0198,  0.0274,  ..., -0.0046, -0.0038, -0.0104]]),\n",
       " 2250: tensor([[-0.0033, -0.0038,  0.0295,  ..., -0.0113,  0.0005, -0.0309]]),\n",
       " 7055: tensor([[-0.0101,  0.0112,  0.0248,  ..., -0.0137, -0.0138, -0.0320]]),\n",
       " 5820: tensor([[-0.0373, -0.0131,  0.0174,  ..., -0.0128, -0.0173, -0.0287]]),\n",
       " 3991: tensor([[ 0.0061, -0.0093,  0.0196,  ..., -0.0302, -0.0105, -0.0226]]),\n",
       " 8738: tensor([[-0.0155,  0.0029,  0.0131,  ..., -0.0201, -0.0357, -0.0133]]),\n",
       " 4018: tensor([[-0.0003,  0.0212,  0.0048,  ...,  0.0048, -0.0258, -0.0255]]),\n",
       " 5942: tensor([[-0.0115,  0.0168,  0.0078,  ..., -0.0122,  0.0048, -0.0076]]),\n",
       " 7137: tensor([[-0.0042,  0.0007,  0.0060,  ..., -0.0067, -0.0113, -0.0189]]),\n",
       " 2332: tensor([[-0.0258,  0.0077, -0.0162,  ..., -0.0082, -0.0240, -0.0159]]),\n",
       " 6440: tensor([[-3.1544e-02,  9.5908e-03,  9.3904e-05,  ...,  4.0483e-03,\n",
       "           1.7493e-03, -2.1472e-02]]),\n",
       " 3645: tensor([[ 0.0163,  0.0215,  0.0437,  ...,  0.0031, -0.0011, -0.0264]]),\n",
       " 10241: tensor([[-0.0009,  0.0164,  0.0190,  ..., -0.0173,  0.0039, -0.0161]]),\n",
       " 5696: tensor([[-0.0155, -0.0054,  0.0027,  ..., -0.0024, -0.0011, -0.0128]]),\n",
       " 739: tensor([[-0.0046,  0.0162,  0.0141,  ..., -0.0076, -0.0068, -0.0305]]),\n",
       " 9023: tensor([[-0.0238,  0.0010, -0.0155,  ..., -0.0200, -0.0084, -0.0133]]),\n",
       " 5703: tensor([[-0.0350,  0.0134,  0.0101,  ..., -0.0145, -0.0257,  0.0037]]),\n",
       " 555: tensor([[-0.0216,  0.0189,  0.0179,  ..., -0.0329, -0.0054, -0.0060]]),\n",
       " 8979: tensor([[-0.0131, -0.0030,  0.0027,  ..., -0.0270, -0.0167, -0.0078]]),\n",
       " 3429: tensor([[-4.0063e-02,  4.1538e-03,  1.8262e-02,  ..., -5.2309e-03,\n",
       "          -6.3403e-05, -2.1788e-02]]),\n",
       " 978: tensor([[ 0.0047, -0.0227,  0.0158,  ..., -0.0280, -0.0105, -0.0242]]),\n",
       " 8554: tensor([[ 0.0044,  0.0006,  0.0192,  ..., -0.0202,  0.0007, -0.0047]]),\n",
       " 4274: tensor([[ 0.0119, -0.0015, -0.0187,  ..., -0.0029, -0.0425, -0.0248]]),\n",
       " 1388: tensor([[-0.0086,  0.0037,  0.0187,  ..., -0.0117, -0.0180, -0.0178]]),\n",
       " 1071: tensor([[-0.0434,  0.0004, -0.0213,  ..., -0.0054, -0.0159, -0.0464]]),\n",
       " 3804: tensor([[-0.0221,  0.0213, -0.0102,  ..., -0.0092, -0.0003, -0.0251]]),\n",
       " 7239: tensor([[-0.0136,  0.0070, -0.0076,  ..., -0.0157, -0.0179, -0.0380]]),\n",
       " 3966: tensor([[ 0.0062,  0.0056,  0.0404,  ..., -0.0155, -0.0263, -0.0426]]),\n",
       " 1113: tensor([[-0.0066,  0.0018, -0.0015,  ..., -0.0176, -0.0243, -0.0124]]),\n",
       " 8436: tensor([[-0.0292,  0.0096,  0.0273,  ...,  0.0008,  0.0053, -0.0295]]),\n",
       " 4316: tensor([[-0.0177, -0.0100,  0.0235,  ..., -0.0053, -0.0063, -0.0363]]),\n",
       " 437: tensor([[-0.0113,  0.0024,  0.0092,  ..., -0.0211, -0.0200, -0.0067]]),\n",
       " 5598: tensor([[-0.0368, -0.0150,  0.0071,  ..., -0.0156, -0.0111,  0.0065]]),\n",
       " 9141: tensor([[-0.0040,  0.0070,  0.0073,  ..., -0.0377,  0.0039, -0.0319]]),\n",
       " 5661: tensor([[ 0.0199, -0.0014,  0.0109,  ..., -0.0204, -0.0207, -0.0126]]),\n",
       " 5888: tensor([[ 0.0010,  0.0306,  0.0395,  ..., -0.0356, -0.0193, -0.0024]]),\n",
       " 927: tensor([[ 0.0002,  0.0014,  0.0099,  ..., -0.0024, -0.0186,  0.0004]]),\n",
       " 2101: tensor([[-0.0296, -0.0312, -0.0319,  ..., -0.0113, -0.0132, -0.0102]]),\n",
       " 7304: tensor([[-0.0360,  0.0040,  0.0127,  ..., -0.0380,  0.0083, -0.0315]]),\n",
       " 10072: tensor([[-0.0361,  0.0002,  0.0155,  ..., -0.0153, -0.0070, -0.0391]]),\n",
       " 3476: tensor([[-0.0107, -0.0008,  0.0075,  ..., -0.0273, -0.0230, -0.0203]]),\n",
       " 6673: tensor([[ 0.0041, -0.0104,  0.0299,  ..., -0.0322, -0.0042, -0.0325]]),\n",
       " 8926: tensor([[-0.0088,  0.0098,  0.0256,  ..., -0.0030, -0.0097, -0.0372]]),\n",
       " 9385: tensor([[-0.0128,  0.0098,  0.0205,  ..., -0.0079, -0.0059, -0.0369]]),\n",
       " 8844: tensor([[-0.0233, -0.0112,  0.0316,  ..., -0.0209, -0.0227,  0.0002]]),\n",
       " 6711: tensor([[-0.0197,  0.0105,  0.0227,  ..., -0.0252, -0.0196, -0.0176]]),\n",
       " 3514: tensor([[ 0.0111,  0.0017,  0.0327,  ..., -0.0576, -0.0079, -0.0147]]),\n",
       " 10110: tensor([[ 0.0149, -0.0100,  0.0300,  ..., -0.0250, -0.0063, -0.0119]]),\n",
       " 791: tensor([[-0.0162, -0.0122, -0.0070,  ..., -0.0073, -0.0251, -0.0320]]),\n",
       " 468: tensor([[-0.0178,  0.0086,  0.0070,  ..., -0.0158, -0.0038, -0.0332]]),\n",
       " 4349: tensor([[ 0.0040, -0.0211,  0.0288,  ..., -0.0209,  0.0067, -0.0080]]),\n",
       " 8469: tensor([[-0.0108,  0.0097,  0.0304,  ..., -0.0266, -0.0114, -0.0204]]),\n",
       " 8790: tensor([[-0.0305, -0.0085,  0.0060,  ..., -0.0355, -0.0122, -0.0342]]),\n",
       " 3939: tensor([[-0.0048, -0.0008,  0.0215,  ..., -0.0063, -0.0128, -0.0162]]),\n",
       " 7266: tensor([[-0.0191, -0.0125,  0.0096,  ...,  0.0079,  0.0085, -0.0291]]),\n",
       " 2063: tensor([[-0.0342,  0.0043,  0.0050,  ..., -0.0076, -0.0239, -0.0119]]),\n",
       " 845: tensor([[-0.0124,  0.0118,  0.0240,  ..., -0.0199, -0.0251, -0.0092]]),\n",
       " 5392: tensor([[-0.0239,  0.0023,  0.0190,  ..., -0.0037, -0.0122, -0.0087]]),\n",
       " 6144: tensor([[-0.0100,  0.0098,  0.0081,  ..., -0.0229, -0.0060, -0.0259]]),\n",
       " 4931: tensor([[-0.0171,  0.0098,  0.0407,  ..., -0.0243, -0.0131, -0.0250]]),\n",
       " 10745: tensor([[-0.0416,  0.0164,  0.0142,  ..., -0.0064, -0.0102, -0.0281]]),\n",
       " 3341: tensor([[-0.0229, -0.0218,  0.0042,  ..., -0.0072, -0.0057, -0.0050]]),\n",
       " 7433: tensor([[-0.0202, -0.0008,  0.0357,  ..., -0.0080, -0.0085, -0.0130]]),\n",
       " 2636: tensor([[-0.0049, -0.0028,  0.0394,  ..., -0.0424, -0.0004, -0.0048]]),\n",
       " 10891: tensor([[-0.0144,  0.0064,  0.0036,  ..., -0.0334,  0.0047, -0.0090]]),\n",
       " 1719: tensor([[-0.0119,  0.0239,  0.0193,  ..., -0.0231, -0.0016, -0.0425]]),\n",
       " 2754: tensor([[-1.2617e-02,  2.1517e-02, -2.0936e-05,  ..., -2.1792e-02,\n",
       "          -2.3837e-03, -2.6805e-02]]),\n",
       " 7551: tensor([[ 0.0022,  0.0040,  0.0270,  ..., -0.0062, -0.0145, -0.0209]]),\n",
       " 4787: tensor([[-0.0174, -0.0029, -0.0182,  ..., -0.0065, -0.0066, -0.0132]]),\n",
       " 1582: tensor([[ 0.0066, -0.0087,  0.0070,  ..., -0.0259, -0.0147, -0.0208]]),\n",
       " 2880: tensor([[-0.0031, -0.0016,  0.0114,  ..., -0.0126,  0.0009, -0.0152]]),\n",
       " 9629: tensor([[ 0.0062,  0.0075,  0.0210,  ..., -0.0167, -0.0243, -0.0144]]),\n",
       " 5109: tensor([[-0.0233, -0.0062, -0.0052,  ..., -0.0163, -0.0115, -0.0231]]),\n",
       " 3223: tensor([[ 0.0042, -0.0042,  0.0162,  ..., -0.0269,  0.0061, -0.0126]]),\n",
       " 10627: tensor([[-0.0185,  0.0073, -0.0012,  ..., -0.0324,  0.0021, -0.0198]]),\n",
       " 4853: tensor([[-0.0223, -0.0135,  0.0109,  ..., -0.0086, -0.0001, -0.0308]]),\n",
       " 6026: tensor([[-0.0205, -0.0096,  0.0137,  ..., -0.0026,  0.0004, -0.0196]]),\n",
       " 1746: tensor([[-0.0050, -0.0175,  0.0229,  ..., -0.0174,  0.0010, -0.0295]]),\n",
       " 4543: tensor([[ 0.0011,  0.0146,  0.0306,  ..., -0.0210, -0.0126, -0.0158]]),\n",
       " 8263: tensor([[-0.0378, -0.0107,  0.0074,  ..., -0.0153, -0.0007, -0.0296]]),\n",
       " 2669: tensor([[-0.0214, -0.0023,  0.0131,  ..., -0.0078, -0.0156, -0.0057]]),\n",
       " 7795: tensor([[-0.0535, -0.0034,  0.0126,  ..., -0.0321, -0.0185,  0.0061]]),\n",
       " 11194: tensor([[-0.0185, -0.0036,  0.0078,  ..., -0.0262,  0.0229, -0.0186]]),\n",
       " 2590: tensor([[-0.0137, -0.0017, -0.0004,  ..., -0.0331, -0.0150, -0.0049]]),\n",
       " 1892: tensor([[ 0.0064, -0.0369,  0.0235,  ..., -0.0112,  0.0008, -0.0230]]),\n",
       " 262: tensor([[-0.0204, -0.0006,  0.0039,  ..., -0.0235, -0.0280, -0.0200]]),\n",
       " 7841: tensor([[ 0.0109,  0.0080,  0.0299,  ...,  0.0183, -0.0085, -0.0346]]),\n",
       " 5034: tensor([[-2.2838e-02,  2.3874e-03,  1.2239e-02,  ..., -3.3188e-02,\n",
       "           8.0910e-05,  1.9037e-03]]),\n",
       " 9714: tensor([[-0.0068,  0.0001,  0.0301,  ..., -0.0169, -0.0124, -0.0121]]),\n",
       " 3185: tensor([[ 0.0171, -0.0233,  0.0367,  ..., -0.0079, -0.0221, -0.0054]]),\n",
       " 6079: tensor([[-0.0070,  0.0110,  0.0191,  ..., -0.0180, -0.0009, -0.0275]]),\n",
       " 10581: tensor([[-0.0376,  0.0162,  0.0161,  ..., -0.0248, -0.0151, -0.0139]]),\n",
       " 10678: tensor([[-0.0047,  0.0151,  0.0295,  ..., -0.0198,  0.0042,  0.0095]]),\n",
       " 6380: tensor([[-0.0103,  0.0052,  0.0239,  ..., -0.0175, -0.0244, -0.0163]]),\n",
       " 5156: tensor([[-0.0156, -0.0096,  0.0221,  ..., -0.0255, -0.0165, -0.0273]]),\n",
       " 9676: tensor([[-0.0197,  0.0095,  0.0215,  ..., -0.0117, -0.0078, -0.0290]]),\n",
       " 7923: tensor([[-0.0265,  0.0178,  0.0373,  ..., -0.0166, -0.0122, -0.0137]]),\n",
       " 300: tensor([[-0.0215, -0.0011,  0.0118,  ..., -0.0339, -0.0161, -0.0097]]),\n",
       " 4421: tensor([[ 0.0092, -0.0097,  0.0372,  ..., -0.0192, -0.0193, -0.0254]]),\n",
       " 8301: tensor([[-0.0238, -0.0014,  0.0030,  ..., -0.0208, -0.0200, -0.0451]]),\n",
       " 1624: tensor([[-0.0130, -0.0120,  0.0066,  ...,  0.0003, -0.0177, -0.0217]]),\n",
       " 35: tensor([[-0.0067, -0.0071,  0.0062,  ..., -0.0169, -0.0123, -0.0161]]),\n",
       " 7762: tensor([[-0.0172, -0.0007,  0.0026,  ..., -0.0126, -0.0092, -0.0206]]),\n",
       " 9837: tensor([[-0.0380,  0.0084, -0.0014,  ..., -0.0168, -0.0410, -0.0358]]),\n",
       " 11163: tensor([[ 0.0015,  0.0015,  0.0458,  ..., -0.0110, -0.0276, -0.0121]]),\n",
       " 2567: tensor([[ 0.0070, -0.0059,  0.0297,  ...,  0.0018,  0.0103, -0.0248]]),\n",
       " 1448: tensor([[-0.0303, -0.0081,  0.0047,  ..., -0.0286, -0.0251, -0.0027]]),\n",
       " 10839: tensor([[-0.0106, -0.0124,  0.0292,  ..., -0.0040, -0.0192, -0.0054]]),\n",
       " 8294: tensor([[-0.0299, -0.0127,  0.0141,  ..., -0.0220, -0.0189,  0.0059]]),\n",
       " 295: tensor([[-0.0115,  0.0130,  0.0083,  ...,  0.0177, -0.0362, -0.0219]]),\n",
       " 6215: tensor([[-0.0314, -0.0012,  0.0125,  ..., -0.0143,  0.0053, -0.0217]]),\n",
       " 1865: tensor([[ 0.0067, -0.0181,  0.0297,  ..., -0.0207, -0.0113, -0.0136]]),\n",
       " 4999: tensor([[-0.0379, -0.0198, -0.0120,  ..., -0.0122, -0.0042, -0.0266]]),\n",
       " 10414: tensor([[-0.0254,  0.0102,  0.0095,  ..., -0.0306, -0.0137, -0.0307]]),\n",
       " 3010: tensor([[-0.0377, -0.0049,  0.0063,  ..., -0.0119, -0.0018, -0.0343]]),\n",
       " 9681: tensor([[-0.0045, -0.0121,  0.0223,  ..., -0.0174, -0.0062, -0.0161]]),\n",
       " 2828: tensor([[-0.0218, -0.0031,  0.0130,  ..., -0.0282, -0.0105, -0.0039]]),\n",
       " 5258: tensor([[-0.0241,  0.0147,  0.0157,  ..., -0.0027,  0.0138, -0.0206]]),\n",
       " 9578: tensor([[-0.0055, -0.0137,  0.0320,  ..., -0.0179, -0.0061, -0.0010]]),\n",
       " 3172: tensor([[-0.0061, -0.0005,  0.0200,  ..., -0.0165, -0.0117, -0.0203]]),\n",
       " 1907: tensor([[-0.0078, -0.0056,  0.0151,  ..., -0.0072, -0.0108, -0.0040]]),\n",
       " 10576: tensor([[ 0.0002, -0.0218,  0.0144,  ...,  0.0094, -0.0026, -0.0136]]),\n",
       " 6377: tensor([[-0.0228, -0.0075, -0.0001,  ..., -0.0254, -0.0011, -0.0288]]),\n",
       " 2405: tensor([[-0.0149,  0.0074,  0.0150,  ..., -0.0355, -0.0240, -0.0052]]),\n",
       " 11001: tensor([[-0.0312, -0.0195, -0.0066,  ..., -0.0111, -0.0072, -0.0117]]),\n",
       " 9955: tensor([[-0.0066,  0.0061,  0.0112,  ..., -0.0103, -0.0153, -0.0005]]),\n",
       " 7600: tensor([[-0.0102,  0.0148,  0.0161,  ..., -0.0335, -0.0213, -0.0230]]),\n",
       " 133: tensor([[-0.0222, -0.0047,  0.0003,  ..., -0.0085, -0.0159, -0.0015]]),\n",
       " 2915: tensor([[-0.0533, -0.0152,  0.0113,  ...,  0.0127, -0.0057, -0.0297]]),\n",
       " 9445: tensor([[-0.0009, -0.0051,  0.0253,  ..., -0.0216, -0.0090, -0.0169]]),\n",
       " 5365: tensor([[-0.0047,  0.0027,  0.0344,  ..., -0.0176, -0.0185, -0.0340]]),\n",
       " 10866: tensor([[-0.0291,  0.0047,  0.0032,  ..., -0.0082, -0.0017, -0.0084]]),\n",
       " 1417: tensor([[-0.0054,  0.0031,  0.0453,  ..., -0.0085, -0.0227, -0.0296]]),\n",
       " 8132: tensor([[-0.0474,  0.0015,  0.0129,  ..., -0.0247,  0.0010, -0.0155]]),\n",
       " 4612: tensor([[-0.0202,  0.0121,  0.0169,  ..., -0.0275, -0.0125, -0.0217]]),\n",
       " 2538: tensor([[-0.0161, -0.0065, -0.0022,  ..., -0.0339, -0.0077,  0.0068]]),\n",
       " 9868: tensor([[ 0.0049, -0.0089,  0.0289,  ..., -0.0137, -0.0089, -0.0175]]),\n",
       " 8050: tensor([[-2.1506e-03,  1.3835e-03,  3.9502e-02,  ..., -8.4533e-05,\n",
       "          -2.6652e-02, -2.2806e-02]]),\n",
       " 4770: tensor([[-0.0198,  0.0120,  0.0181,  ..., -0.0001, -0.0144, -0.0181]]),\n",
       " 10904: tensor([[-0.0370,  0.0026,  0.0246,  ..., -0.0246, -0.0343, -0.0158]]),\n",
       " 4489: tensor([[-0.0132,  0.0112,  0.0153,  ..., -0.0103, -0.0234, -0.0357]]),\n",
       " 1575: tensor([[ 0.0076, -0.0047,  0.0124,  ...,  0.0020, -0.0410, -0.0241]]),\n",
       " 6328: tensor([[-6.4308e-05,  1.5980e-03,  2.3911e-02,  ...,  9.2716e-03,\n",
       "           4.5547e-04, -2.6081e-02]]),\n",
       " 10529: tensor([[-0.0145,  0.0058,  0.0089,  ..., -0.0113,  0.0140, -0.0257]]),\n",
       " 1958: tensor([[ 0.0262,  0.0225,  0.0168,  ...,  0.0028, -0.0059, -0.0318]]),\n",
       " 9527: tensor([[ 0.0124, -0.0055,  0.0335,  ..., -0.0185, -0.0041, -0.0145]]),\n",
       " 5207: tensor([[-0.0232, -0.0084,  0.0336,  ..., -0.0201, -0.0225, -0.0220]]),\n",
       " 2877: tensor([[-0.0178,  0.0038,  0.0290,  ..., -0.0228, -0.0072, -0.0175]]),\n",
       " 3938: tensor([[ 1.5863e-02,  3.5158e-03,  1.5888e-02,  ..., -1.9717e-02,\n",
       "          -2.1121e-02, -7.1934e-05]]),\n",
       " 8791: tensor([[-0.0281, -0.0206,  0.0124,  ..., -0.0187, -0.0091, -0.0220]]),\n",
       " 8468: tensor([[-0.0324, -0.0146,  0.0077,  ..., -0.0176, -0.0047, -0.0065]]),\n",
       " 4348: tensor([[ 1.8551e-02,  1.4295e-02,  2.5731e-02,  ..., -5.0197e-03,\n",
       "          -3.6386e-05,  3.6717e-03]]),\n",
       " 2062: tensor([[ 0.0248, -0.0010,  0.0158,  ..., -0.0236, -0.0170, -0.0169]]),\n",
       " 844: tensor([[ 0.0038,  0.0061,  0.0269,  ..., -0.0088, -0.0298, -0.0439]]),\n",
       " 7267: tensor([[-0.0184, -0.0141,  0.0219,  ..., -0.0338, -0.0206, -0.0189]]),\n",
       " 10111: tensor([[-0.0427,  0.0015,  0.0080,  ..., -0.0274, -0.0147, -0.0177]]),\n",
       " 3515: tensor([[-0.0140,  0.0081, -0.0011,  ..., -0.0045,  0.0079, -0.0360]]),\n",
       " 6710: tensor([[-0.0208,  0.0070,  0.0016,  ..., -0.0125, -0.0260, -0.0411]]),\n",
       " 8845: tensor([[-0.0051, -0.0047,  0.0012,  ..., -0.0281, -0.0305, -0.0071]]),\n",
       " 469: tensor([[ 0.0143, -0.0099,  0.0181,  ...,  0.0102, -0.0077, -0.0411]]),\n",
       " 790: tensor([[ 0.0045,  0.0071,  0.0508,  ..., -0.0201, -0.0127, -0.0215]]),\n",
       " 8927: tensor([[-0.0369, -0.0163,  0.0058,  ..., -0.0383, -0.0197, -0.0026]]),\n",
       " 6672: tensor([[-0.0188, -0.0082,  0.0282,  ..., -0.0157, -0.0023, -0.0039]]),\n",
       " 3477: tensor([[-0.0258,  0.0068,  0.0043,  ..., -0.0317, -0.0131, -0.0233]]),\n",
       " 10073: tensor([[-0.0207, -0.0067,  0.0222,  ...,  0.0056,  0.0020, -0.0174]]),\n",
       " 9384: tensor([[-0.0136,  0.0273,  0.0301,  ..., -0.0265, -0.0050, -0.0338]]),\n",
       " 7305: tensor([[ 0.0210,  0.0188,  0.0335,  ..., -0.0107, -0.0016, -0.0162]]),\n",
       " 926: tensor([[-0.0272, -0.0075,  0.0003,  ..., -0.0086, -0.0218, -0.0129]]),\n",
       " 2100: tensor([[-0.0046,  0.0196,  0.0281,  ..., -0.0110, -0.0216, -0.0088]]),\n",
       " 5889: tensor([[-0.0127,  0.0082,  0.0333,  ..., -0.0361,  0.0049,  0.0160]]),\n",
       " 5660: tensor([[-0.0074, -0.0201,  0.0262,  ..., -0.0229,  0.0014, -0.0272]]),\n",
       " 9140: tensor([[ 0.0152,  0.0012,  0.0294,  ..., -0.0278,  0.0115, -0.0139]]),\n",
       " 5599: tensor([[-0.0415, -0.0086, -0.0044,  ..., -0.0074, -0.0271, -0.0103]]),\n",
       " 436: tensor([[-0.0035,  0.0143,  0.0200,  ..., -0.0103, -0.0260, -0.0106]]),\n",
       " 7238: tensor([[-0.0034, -0.0016, -0.0049,  ..., -0.0061, -0.0148, -0.0359]]),\n",
       " 4317: tensor([[ 0.0129,  0.0106,  0.0280,  ..., -0.0264, -0.0135, -0.0403]]),\n",
       " 8437: tensor([[-0.0307, -0.0150,  0.0255,  ..., -0.0193, -0.0091, -0.0316]]),\n",
       " 1112: tensor([[-0.0440, -0.0154, -0.0001,  ..., -0.0216, -0.0117, -0.0032]]),\n",
       " 3967: tensor([[-0.0147,  0.0207,  0.0191,  ..., -0.0128, -0.0192, -0.0249]]),\n",
       " 979: tensor([[-0.0533,  0.0040, -0.0041,  ..., -0.0123, -0.0167, -0.0095]]),\n",
       " 3805: tensor([[-0.0063, -0.0205,  0.0166,  ..., -0.0170, -0.0011, -0.0361]]),\n",
       " 1070: tensor([[ 0.0041,  0.0089,  0.0125,  ..., -0.0073, -0.0170, -0.0250]]),\n",
       " 4275: tensor([[-0.0116,  0.0009,  0.0287,  ..., -0.0232, -0.0241, -0.0232]]),\n",
       " 8555: tensor([[-0.0361, -0.0181, -0.0137,  ..., -0.0112, -0.0041, -0.0252]]),\n",
       " 1389: tensor([[-0.0345, -0.0162,  0.0119,  ..., -0.0088, -0.0345, -0.0186]]),\n",
       " 554: tensor([[-0.0117,  0.0161,  0.0210,  ..., -0.0037, -0.0192, -0.0321]]),\n",
       " 5702: tensor([[ 0.0003,  0.0072,  0.0015,  ..., -0.0278, -0.0091, -0.0372]]),\n",
       " 9022: tensor([[-0.0387,  0.0056,  0.0420,  ..., -0.0296, -0.0266, -0.0357]]),\n",
       " 3428: tensor([[-0.0093,  0.0200,  0.0194,  ..., -0.0181, -0.0102, -0.0326]]),\n",
       " 8978: tensor([[-0.0112,  0.0168,  0.0161,  ..., -0.0245, -0.0140, -0.0347]]),\n",
       " 10240: tensor([[-0.0290, -0.0011,  0.0227,  ..., -0.0090, -0.0064, -0.0128]]),\n",
       " 3644: tensor([[ 0.0053,  0.0160,  0.0453,  ..., -0.0145, -0.0008, -0.0287]]),\n",
       " 6441: tensor([[-0.0068,  0.0152,  0.0293,  ..., -0.0190, -0.0110, -0.0182]]),\n",
       " 738: tensor([[-0.0178, -0.0263,  0.0333,  ..., -0.0070, -0.0285, -0.0252]]),\n",
       " 5697: tensor([[-0.0029, -0.0085,  0.0309,  ..., -0.0007,  0.0050, -0.0106]]),\n",
       " 4019: tensor([[-0.0223, -0.0002, -0.0030,  ...,  0.0102, -0.0124, -0.0070]]),\n",
       " 8739: tensor([[-0.0307, -0.0113,  0.0057,  ..., -0.0296, -0.0032, -0.0391]]),\n",
       " 3990: tensor([[-0.0176,  0.0013,  0.0134,  ..., -0.0490, -0.0063, -0.0093]]),\n",
       " 2333: tensor([[-0.0216, -0.0244, -0.0015,  ..., -0.0285, -0.0018,  0.0001]]),\n",
       " 7136: tensor([[-0.0398, -0.0136,  0.0332,  ..., -0.0300, -0.0122, -0.0261]]),\n",
       " 5943: tensor([[-0.0229, -0.0078,  0.0141,  ..., -0.0276, -0.0182, -0.0287]]),\n",
       " 1087: tensor([[ 0.0125,  0.0033, -0.0112,  ..., -0.0241, -0.0323, -0.0413]]),\n",
       " 4282: tensor([[-0.0137,  0.0094,  0.0098,  ..., -0.0272,  0.0098,  0.0145]]),\n",
       " 5821: tensor([[-0.0330, -0.0102,  0.0191,  ..., -0.0238, -0.0177,  0.0043]]),\n",
       " 7054: tensor([[-0.0277, -0.0103,  0.0131,  ..., -0.0102, -0.0013, -0.0127]]),\n",
       " 2251: tensor([[-0.0233, -0.0226,  0.0060,  ..., -0.0054, -0.0096, -0.0112]]),\n",
       " 6523: tensor([[-0.0396, -0.0044,  0.0169,  ..., -0.0079,  0.0048, -0.0193]]),\n",
       " 3726: tensor([[-0.0127,  0.0088,  0.0167,  ..., -0.0094, -0.0077, -0.0227]]),\n",
       " 10322: tensor([[-5.2865e-03,  7.0489e-05,  2.1628e-02,  ..., -1.7863e-02,\n",
       "          -7.6546e-03, -2.3674e-02]]),\n",
       " 2095: tensor([[-0.0468, -0.0101,  0.0029,  ..., -0.0267, -0.0036, -0.0017]]),\n",
       " 7169: tensor([[-0.0099, -0.0005,  0.0157,  ..., -0.0435, -0.0101, -0.0206]]),\n",
       " 7290: tensor([[-0.0083, -0.0055,  0.0242,  ..., -0.0098, -0.0316, -0.0134]]),\n",
       " 6833: tensor([[-0.0018,  0.0159,  0.0132,  ...,  0.0027, -0.0006, -0.0332]]),\n",
       " 8766: tensor([[ 0.0191,  0.0170,  0.0250,  ..., -0.0101, -0.0146, -0.0189]]),\n",
       " 4046: tensor([[ 7.8876e-05, -2.7893e-02,  1.9602e-02,  ..., -2.3944e-02,\n",
       "          -3.2094e-02, -1.8787e-02]]),\n",
       " 1243: tensor([[ 0.0054,  0.0081,  0.0310,  ..., -0.0445, -0.0257, -0.0134]]),\n",
       " 9211: tensor([[-0.0009, -0.0033,  0.0382,  ..., -0.0138, -0.0042, -0.0260]]),\n",
       " 5531: tensor([[ 0.0148,  0.0021,  0.0443,  ..., -0.0206,  0.0071, -0.0238]]),\n",
       " 767: tensor([[ 0.0062,  0.0027,  0.0136,  ..., -0.0025, -0.0070, -0.0457]]),\n",
       " 605: tensor([[-0.0302,  0.0110,  0.0178,  ..., -0.0141, -0.0164, -0.0119]]),\n",
       " 9373: tensor([[-0.0118,  0.0162,  0.0202,  ..., -0.0106, -0.0034, -0.0052]]),\n",
       " 5453: tensor([[-0.0029, -0.0069,  0.0357,  ..., -0.0343, -0.0151, -0.0253]]),\n",
       " 3779: tensor([[-0.0195, -0.0344,  0.0188,  ..., -0.0167, -0.0032, -0.0240]]),\n",
       " 6685: tensor([[-0.0136,  0.0057,  0.0250,  ..., -0.0078, -0.0160,  0.0033]]),\n",
       " 3480: tensor([[-0.0040,  0.0197,  0.0115,  ..., -0.0230, -0.0242, -0.0223]]),\n",
       " 10084: tensor([[-0.0337,  0.0005,  0.0201,  ..., -0.0225, -0.0335, -0.0085]]),\n",
       " 1321: tensor([[-0.0095, -0.0158,  0.0342,  ..., -0.0005, -0.0037, -0.0389]]),\n",
       " 8604: tensor([[-0.0365, -0.0039, -0.0013,  ..., -0.0041, -0.0100, -0.0229]]),\n",
       " 4124: tensor([[ 0.0017, -0.0041,  0.0259,  ..., -0.0191, -0.0141, -0.0287]]),\n",
       " 6951: tensor([[-0.0387,  0.0074,  0.0170,  ..., -0.0198, -0.0242, -0.0183]]),\n",
       " 10528: tensor([[-0.0400,  0.0038,  0.0159,  ..., -0.0195, -0.0117, -0.0296]]),\n",
       " 1959: tensor([[-0.0177,  0.0025,  0.0099,  ..., -0.0139, -0.0224, -0.0193]]),\n",
       " 6329: tensor([[-0.0176,  0.0088,  0.0153,  ..., -0.0293, -0.0174, -0.0247]]),\n",
       " 2876: tensor([[-0.0182,  0.0100,  0.0187,  ..., -0.0198, -0.0185, -0.0259]]),\n",
       " 5206: tensor([[-0.0601, -0.0061,  0.0097,  ...,  0.0007, -0.0143, -0.0190]]),\n",
       " 9526: tensor([[ 0.0032, -0.0113,  0.0239,  ..., -0.0052, -0.0122, -0.0266]]),\n",
       " 4488: tensor([[-0.0339, -0.0112, -0.0063,  ..., -0.0213, -0.0226, -0.0084]]),\n",
       " 10905: tensor([[-0.0149, -0.0179,  0.0143,  ..., -0.0074, -0.0103, -0.0280]]),\n",
       " 1574: tensor([[ 0.0055, -0.0177,  0.0246,  ...,  0.0063, -0.0015, -0.0428]]),\n",
       " 4771: tensor([[ 0.0186,  0.0074,  0.0315,  ..., -0.0082, -0.0071, -0.0183]]),\n",
       " 8051: tensor([[-0.0302,  0.0008,  0.0123,  ..., -0.0096, -0.0164, -0.0451]]),\n",
       " 4613: tensor([[-0.0124,  0.0028,  0.0025,  ..., -0.0088, -0.0018, -0.0371]]),\n",
       " 8133: tensor([[-0.0115, -0.0131, -0.0063,  ..., -0.0391, -0.0118, -0.0013]]),\n",
       " 10867: tensor([[-0.0246,  0.0035, -0.0020,  ..., -0.0060,  0.0017, -0.0124]]),\n",
       " 1416: tensor([[ 0.0072, -0.0049,  0.0292,  ..., -0.0081, -0.0337, -0.0257]]),\n",
       " 9869: tensor([[ 0.0197, -0.0331,  0.0276,  ..., -0.0214, -0.0014, -0.0145]]),\n",
       " 2539: tensor([[-0.0344, -0.0023, -0.0063,  ..., -0.0245, -0.0086, -0.0150]]),\n",
       " 5364: tensor([[-0.0512, -0.0044, -0.0015,  ..., -0.0025,  0.0030, -0.0204]]),\n",
       " 9444: tensor([[-0.0260, -0.0175,  0.0316,  ..., -0.0179, -0.0061, -0.0098]]),\n",
       " 132: tensor([[ 0.0047, -0.0215,  0.0178,  ..., -0.0205,  0.0075, -0.0248]]),\n",
       " 2914: tensor([[-0.0101,  0.0180, -0.0050,  ...,  0.0080,  0.0066, -0.0275]]),\n",
       " 7601: tensor([[-0.0256, -0.0004,  0.0103,  ..., -0.0249, -0.0319, -0.0059]]),\n",
       " 9954: tensor([[-0.0253,  0.0134,  0.0055,  ..., -0.0275,  0.0166,  0.0020]]),\n",
       " 11000: tensor([[-0.0064, -0.0134,  0.0317,  ..., -0.0075, -0.0033, -0.0257]]),\n",
       " 2404: tensor([[-0.0201, -0.0012,  0.0228,  ..., -0.0177, -0.0211, -0.0431]]),\n",
       " 9579: tensor([[-0.0057, -0.0074,  0.0389,  ..., -0.0347, -0.0263, -0.0005]]),\n",
       " 5259: tensor([[ 0.0172, -0.0204,  0.0338,  ..., -0.0102, -0.0125, -0.0243]]),\n",
       " 2829: tensor([[ 0.0066, -0.0091,  0.0227,  ..., -0.0306, -0.0171, -0.0091]]),\n",
       " 9680: tensor([[-0.0323, -0.0049, -0.0044,  ..., -0.0145, -0.0150, -0.0271]]),\n",
       " 6376: tensor([[-0.0169, -0.0058,  0.0138,  ..., -0.0180,  0.0059, -0.0356]]),\n",
       " 1906: tensor([[-0.0161,  0.0086,  0.0151,  ..., -0.0203, -0.0209, -0.0273]]),\n",
       " 10577: tensor([[-0.0113, -0.0237,  0.0228,  ..., -0.0137, -0.0087, -0.0102]]),\n",
       " 3173: tensor([[-0.0226, -0.0042,  0.0080,  ..., -0.0166, -0.0130, -0.0074]]),\n",
       " 294: tensor([[-0.0240,  0.0013,  0.0140,  ..., -0.0318, -0.0058, -0.0186]]),\n",
       " 3011: tensor([[-0.0230, -0.0014, -0.0001,  ..., -0.0460, -0.0087, -0.0227]]),\n",
       " 1864: tensor([[-0.0319, -0.0070,  0.0011,  ..., -0.0057, -0.0070, -0.0183]]),\n",
       " 10415: tensor([[ 0.0127,  0.0184,  0.0229,  ..., -0.0066, -0.0216, -0.0254]]),\n",
       " 4998: tensor([[-0.0144, -0.0012,  0.0126,  ..., -0.0234,  0.0017, -0.0302]]),\n",
       " 6214: tensor([[ 0.0078, -0.0069,  0.0198,  ..., -0.0221, -0.0056, -0.0139]]),\n",
       " 2566: tensor([[-0.0031, -0.0004,  0.0263,  ..., -0.0302, -0.0245, -0.0044]]),\n",
       " 11162: tensor([[-0.0421, -0.0230,  0.0176,  ..., -0.0197, -0.0132, -0.0297]]),\n",
       " 9836: tensor([[-0.0174, -0.0051, -0.0135,  ..., -0.0359, -0.0098, -0.0226]]),\n",
       " 7763: tensor([[-0.0341,  0.0062,  0.0201,  ..., -0.0294, -0.0131, -0.0353]]),\n",
       " 1449: tensor([[-0.0078, -0.0018,  0.0163,  ...,  0.0016, -0.0109, -0.0186]]),\n",
       " 10838: tensor([[-0.0213, -0.0045,  0.0318,  ..., -0.0186, -0.0017, -0.0163]]),\n",
       " 8295: tensor([[-0.0271, -0.0090,  0.0196,  ..., -0.0290, -0.0176, -0.0281]]),\n",
       " 1625: tensor([[-0.0078, -0.0249,  0.0151,  ..., -0.0136, -0.0120, -0.0167]]),\n",
       " 34: tensor([[ 0.0032, -0.0076,  0.0176,  ..., -0.0114, -0.0056, -0.0135]]),\n",
       " 8300: tensor([[-0.0358,  0.0101,  0.0047,  ..., -0.0046, -0.0077, -0.0228]]),\n",
       " 4420: tensor([[-0.0061, -0.0020,  0.0252,  ..., -0.0035, -0.0257, -0.0338]]),\n",
       " 6381: tensor([[ 0.0125, -0.0055,  0.0227,  ..., -0.0091,  0.0070, -0.0160]]),\n",
       " 10679: tensor([[-0.0049,  0.0118,  0.0234,  ..., -0.0064, -0.0106, -0.0275]]),\n",
       " 10580: tensor([[-0.0257, -0.0090,  0.0207,  ..., -0.0031, -0.0320, -0.0282]]),\n",
       " 3184: tensor([[-0.0117,  0.0043,  0.0148,  ..., -0.0206, -0.0203, -0.0133]]),\n",
       " 6078: tensor([[-0.0204, -0.0115,  0.0160,  ..., -0.0270, -0.0068, -0.0019]]),\n",
       " 301: tensor([[-0.0389, -0.0161,  0.0094,  ...,  0.0053, -0.0084,  0.0028]]),\n",
       " 7922: tensor([[ 0.0006,  0.0084,  0.0222,  ..., -0.0191, -0.0080, -0.0366]]),\n",
       " 9677: tensor([[-0.0391, -0.0028,  0.0120,  ..., -0.0318, -0.0088, -0.0182]]),\n",
       " 5157: tensor([[-0.0083,  0.0038,  0.0484,  ..., -0.0021, -0.0126, -0.0264]]),\n",
       " 1893: tensor([[-0.0167, -0.0300,  0.0212,  ..., -0.0049,  0.0103, -0.0332]]),\n",
       " 9715: tensor([[-0.0421,  0.0096, -0.0070,  ..., -0.0183, -0.0085, -0.0240]]),\n",
       " 5035: tensor([[ 0.0135, -0.0152,  0.0196,  ...,  0.0002, -0.0172, -0.0211]]),\n",
       " 7840: tensor([[-0.0206, -0.0075,  0.0228,  ..., -0.0066, -0.0016, -0.0297]]),\n",
       " 263: tensor([[-0.0455, -0.0012, -0.0076,  ..., -0.0150, -0.0115, -0.0194]]),\n",
       " 8262: tensor([[-0.0040, -0.0248,  0.0245,  ..., -0.0117, -0.0179,  0.0021]]),\n",
       " 4542: tensor([[-0.0264, -0.0104,  0.0091,  ..., -0.0264, -0.0253, -0.0286]]),\n",
       " 1747: tensor([[-0.0442, -0.0188, -0.0050,  ..., -0.0126,  0.0088, -0.0022]]),\n",
       " 2591: tensor([[-0.0181, -0.0045,  0.0056,  ..., -0.0131, -0.0238, -0.0094]]),\n",
       " 11195: tensor([[ 0.0031,  0.0058,  0.0209,  ..., -0.0231, -0.0199, -0.0149]]),\n",
       " 2668: tensor([[-0.0347,  0.0180,  0.0135,  ...,  0.0176, -0.0249, -0.0223]]),\n",
       " 7794: tensor([[-0.0130, -0.0068,  0.0004,  ..., -0.0010,  0.0005, -0.0131]]),\n",
       " 5108: tensor([[-0.0207, -0.0108, -0.0030,  ..., -0.0177, -0.0030, -0.0309]]),\n",
       " 9628: tensor([[-0.0342, -0.0105, -0.0038,  ...,  0.0036, -0.0090, -0.0261]]),\n",
       " 2881: tensor([[ 1.4016e-02,  8.4547e-03,  2.4417e-02,  ...,  3.1906e-05,\n",
       "          -1.8790e-02, -2.5777e-02]]),\n",
       " 6027: tensor([[-0.0382, -0.0027, -0.0081,  ..., -0.0082, -0.0094, -0.0283]]),\n",
       " 4852: tensor([[-0.0060, -0.0109,  0.0152,  ..., -0.0127,  0.0011, -0.0110]]),\n",
       " 10626: tensor([[-0.0292,  0.0069,  0.0047,  ..., -0.0240,  0.0077, -0.0296]]),\n",
       " 3222: tensor([[-0.0063, -0.0075,  0.0018,  ..., -0.0137, -0.0174, -0.0066]]),\n",
       " 7550: tensor([[-0.0225,  0.0103,  0.0101,  ...,  0.0034, -0.0151, -0.0197]]),\n",
       " 2755: tensor([[-0.0193,  0.0010,  0.0057,  ..., -0.0279, -0.0138, -0.0026]]),\n",
       " 1583: tensor([[-0.0216, -0.0035,  0.0238,  ..., -0.0374, -0.0145, -0.0165]]),\n",
       " 4786: tensor([[-0.0413,  0.0099, -0.0013,  ..., -0.0176, -0.0098, -0.0242]]),\n",
       " 2637: tensor([[-1.6846e-02, -3.6519e-06,  2.3528e-02,  ..., -2.7720e-02,\n",
       "          -1.8667e-02, -3.9210e-02]]),\n",
       " 7432: tensor([[-0.0206,  0.0204,  0.0211,  ..., -0.0160, -0.0083, -0.0101]]),\n",
       " 1718: tensor([[-0.0133,  0.0006,  0.0037,  ..., -0.0130, -0.0145, -0.0342]]),\n",
       " 10890: tensor([[-0.0243, -0.0007,  0.0012,  ..., -0.0279, -0.0246, -0.0290]]),\n",
       " 5393: tensor([[-0.0161, -0.0028,  0.0005,  ..., -0.0210, -0.0173, -0.0065]]),\n",
       " 3340: tensor([[-0.0058,  0.0045,  0.0141,  ..., -0.0126, -0.0266, -0.0447]]),\n",
       " 10744: tensor([[-0.0087,  0.0032,  0.0126,  ..., -0.0145,  0.0010, -0.0235]]),\n",
       " 4930: tensor([[ 0.0019, -0.0159,  0.0239,  ..., -0.0260, -0.0080, -0.0148]]),\n",
       " 6145: tensor([[-0.0336, -0.0138,  0.0042,  ...,  0.0028, -0.0238, -0.0264]]),\n",
       " 9556: tensor([[-0.0266,  0.0160,  0.0119,  ..., -0.0332,  0.0035, -0.0333]]),\n",
       " 5276: tensor([[-0.0249, -0.0171, -0.0046,  ..., -0.0148, -0.0331, -0.0007]]),\n",
       " 2806: tensor([[-0.0253,  0.0025,  0.0157,  ..., -0.0097, -0.0120, -0.0190]]),\n",
       " 6359: tensor([[-0.0121,  0.0085,  0.0228,  ..., -0.0089, -0.0112, -0.0441]]),\n",
       " 10558: tensor([[-0.0372, -0.0042,  0.0059,  ..., -0.0126,  0.0089, -0.0099]]),\n",
       " 1929: tensor([[-0.0071, -0.0245,  0.0167,  ..., -0.0014, -0.0072, -0.0162]]),\n",
       " 8021: tensor([[-0.0301, -0.0062,  0.0052,  ..., -0.0185,  0.0231, -0.0267]]),\n",
       " 4701: tensor([[-0.0150,  0.0067,  0.0085,  ..., -0.0185, -0.0103, -0.0030]]),\n",
       " 10975: tensor([[-0.0315, -0.0001,  0.0382,  ..., -0.0209, -0.0104, -0.0328]]),\n",
       " 1504: tensor([[-0.0284, -0.0047,  0.0022,  ..., -0.0261, -0.0080, -0.0111]]),\n",
       " 2549: tensor([[-0.0462,  0.0265, -0.0033,  ..., -0.0363, -0.0099, -0.0108]]),\n",
       " 9819: tensor([[-0.0289,  0.0091,  0.0223,  ..., -0.0209, -0.0175, -0.0333]]),\n",
       " 10817: tensor([[-0.0271,  0.0005, -0.0042,  ..., -0.0145, -0.0207, -0.0249]]),\n",
       " 1466: tensor([[ 0.0177,  0.0082,  0.0281,  ..., -0.0007, -0.0251, -0.0250]]),\n",
       " 8143: tensor([[-0.0379,  0.0032,  0.0163,  ..., -0.0114, -0.0099, -0.0331]]),\n",
       " 4663: tensor([[ 0.0024, -0.0058,  0.0278,  ..., -0.0101, -0.0067, -0.0194]]),\n",
       " 7898: tensor([[-0.0271,  0.0020,  0.0138,  ..., -0.0236, -0.0170, -0.0216]]),\n",
       " 2964: tensor([[-0.0134, -0.0060,  0.0342,  ..., -0.0135, -0.0037, -0.0200]]),\n",
       " 142: tensor([[-0.0090,  0.0050,  0.0235,  ...,  0.0077, -0.0058, -0.0242]]),\n",
       " 9434: tensor([[-0.0245, -0.0001,  0.0214,  ..., -0.0286,  0.0054, -0.0256]]),\n",
       " 5314: tensor([[-0.0032,  0.0069,  0.0213,  ..., -0.0314, -0.0245, -0.0121]]),\n",
       " 8387: tensor([[-0.0062,  0.0067,  0.0239,  ..., -0.0205,  0.0095, -0.0178]]),\n",
       " 11070: tensor([[-0.0368,  0.0049, -0.0020,  ..., -0.0160, -0.0137, -0.0250]]),\n",
       " 2474: tensor([[-0.0212, -0.0045,  0.0287,  ..., -0.0248, -0.0203, -0.0085]]),\n",
       " 7588: tensor([[-0.0413,  0.0144,  0.0240,  ..., -0.0168, -0.0144, -0.0479]]),\n",
       " 7671: tensor([[-0.0209,  0.0123,  0.0218,  ..., -0.0429, -0.0229, -0.0168]]),\n",
       " 9924: tensor([[-0.0265, -0.0109, -0.0003,  ...,  0.0009,  0.0022, -0.0144]]),\n",
       " 1976: tensor([[ 3.1205e-05, -2.0946e-02,  2.0163e-02,  ..., -1.4112e-02,\n",
       "          -7.9819e-03,  1.4805e-04]]),\n",
       " 10507: tensor([[ 0.0123,  0.0172,  0.0205,  ..., -0.0380, -0.0196, -0.0421]]),\n",
       " 3103: tensor([[-0.0036,  0.0013, -0.0056,  ..., -0.0238, -0.0146, -0.0325]]),\n",
       " 6306: tensor([[-0.0215,  0.0270,  0.0291,  ...,  0.0019, -0.0234, -0.0215]]),\n",
       " 2859: tensor([[-0.0257, -0.0119, -0.0046,  ..., -0.0223, -0.0233, -0.0084]]),\n",
       " 5229: tensor([[ 0.0035,  0.0089,  0.0061,  ..., -0.0089, -0.0002, -0.0284]]),\n",
       " 9509: tensor([[ 0.0085, -0.0081,  0.0381,  ..., -0.0066, -0.0098, -0.0098]]),\n",
       " 386: tensor([[-0.0145,  0.0105,  0.0063,  ..., -0.0062, -0.0136, -0.0159]]),\n",
       " 3398: tensor([[ 0.0060, -0.0011,  0.0488,  ...,  0.0052,  0.0019, -0.0115]]),\n",
       " 6264: tensor([[-0.0123,  0.0090,  0.0213,  ..., -0.0126, -0.0169, -0.0363]]),\n",
       " 3061: tensor([[-0.0264,  0.0036,  0.0162,  ..., -0.0111, -0.0261, -0.0013]]),\n",
       " 1814: tensor([[-0.0181,  0.0040, -0.0027,  ..., -0.0169, -0.0102,  0.0005]]),\n",
       " 10465: tensor([[-0.0269, -0.0050,  0.0019,  ..., -0.0094, -0.0103, -0.0131]]),\n",
       " 9792: tensor([[ 0.0018, -0.0157,  0.0227,  ..., -0.0203, -0.0193, -0.0265]]),\n",
       " 1439: tensor([[-0.0118,  0.0071,  0.0210,  ..., -0.0113, -0.0237, -0.0146]]),\n",
       " 10848: tensor([[-0.0178,  0.0272,  0.0116,  ..., -0.0038, -0.0007, -0.0049]]),\n",
       " 9846: tensor([[-0.0110,  0.0083,  0.0205,  ..., -0.0262, -0.0006, -0.0233]]),\n",
       " 7713: tensor([[ 0.0061, -0.0101,  0.0383,  ..., -0.0267, -0.0263, -0.0260]]),\n",
       " 2516: tensor([[-0.0072,  0.0157,  0.0116,  ..., -0.0068, -0.0078, -0.0374]]),\n",
       " 11112: tensor([[ 0.0155,  0.0035,  0.0346,  ..., -0.0214,  0.0035, -0.0066]]),\n",
       " 11087: tensor([[-0.0232, -0.0009,  0.0090,  ..., -0.0240, -0.0273, -0.0178]]),\n",
       " 2483: tensor([[-0.0133,  0.0064,  0.0120,  ..., -0.0186,  0.0039,  0.0126]]),\n",
       " 7686: tensor([[-0.0267,  0.0084,  0.0226,  ..., -0.0152, -0.0022, -0.0249]]),\n",
       " 4450: tensor([[-0.0133, -0.0105,  0.0350,  ..., -0.0141, -0.0265,  0.0048]]),\n",
       " 8370: tensor([[-0.0287,  0.0016,  0.0207,  ..., -0.0285, -0.0148, -0.0189]]),\n",
       " 1655: tensor([[ 0.0194,  0.0245,  0.0259,  ..., -0.0118, -0.0233, -0.0374]]),\n",
       " 44: tensor([[-0.0116, -0.0071,  0.0026,  ..., -0.0047, -0.0238, -0.0352]]),\n",
       " 8089: tensor([[-0.0100, -0.0023,  0.0207,  ..., -0.0122, -0.0081, -0.0161]]),\n",
       " 7952: tensor([[-0.0317, -0.0167,  0.0172,  ..., -0.0341, -0.0142, -0.0097]]),\n",
       " 5127: tensor([[ 0.0155,  0.0073,  0.0249,  ..., -0.0144,  0.0023, -0.0056]]),\n",
       " 9607: tensor([[-0.0246, -0.0239,  0.0018,  ..., -0.0111,  0.0120, -0.0086]]),\n",
       " 371: tensor([[-0.0300,  0.0084,  0.0013,  ..., -0.0249,  0.0042, -0.0245]]),\n",
       " 1981: tensor([[-0.0087,  0.0042,  0.0205,  ..., -0.0305,  0.0048, -0.0182]]),\n",
       " 6008: tensor([[-0.0216, -0.0005,  0.0061,  ..., -0.0147, -0.0278, -0.0163]]),\n",
       " 10609: tensor([[-0.0090,  0.0123,  0.0150,  ..., -0.0135, -0.0005, -0.0436]]),\n",
       " 213: tensor([[-0.0033,  0.0082,  0.0244,  ..., -0.0257, -0.0319, -0.0300]]),\n",
       " 5045: tensor([[-0.0061, -0.0106,  0.0206,  ..., -0.0144, -0.0372, -0.0082]]),\n",
       " 9765: tensor([[ 0.0051,  0.0113,  0.0321,  ..., -0.0231, -0.0233, -0.0204]]),\n",
       " 7830: tensor([[-0.0079,  0.0076,  0.0271,  ...,  0.0011, -0.0005, -0.0328]]),\n",
       " 6293: tensor([[-0.0054, -0.0034,  0.0160,  ..., -0.0057, -0.0052, -0.0214]]),\n",
       " 3096: tensor([[-0.0196,  0.0092,  0.0132,  ..., -0.0179,  0.0013, -0.0435]]),\n",
       " 10492: tensor([[-0.0103,  0.0053,  0.0301,  ..., -0.0322, -0.0206, -0.0401]]),\n",
       " 2618: tensor([[-0.0062, -0.0084, -0.0031,  ..., -0.0268, -0.0069, -0.0094]]),\n",
       " 1737: tensor([[-1.5890e-03, -7.3855e-03,  3.6437e-02,  ..., -1.6940e-02,\n",
       "          -2.0187e-05, -2.2518e-02]]),\n",
       " 4532: tensor([[-0.0181, -0.0008,  0.0407,  ...,  0.0033, -0.0233, -0.0173]]),\n",
       " 8212: tensor([[-0.0180,  0.0131,  0.0164,  ..., -0.0172, -0.0096, -0.0129]]),\n",
       " 10656: tensor([[-0.0143, -0.0307,  0.0160,  ..., -0.0309, -0.0220, -0.0120]]),\n",
       " 3252: tensor([[-0.0132, -0.0079,  0.0305,  ..., -0.0218, -0.0212, -0.0172]]),\n",
       " 6057: tensor([[-0.0183, -0.0171,  0.0198,  ..., -0.0421, -0.0104, -0.0430]]),\n",
       " 4822: tensor([[ 0.0008, -0.0085,  0.0033,  ..., -0.0062, -0.0048, -0.0170]]),\n",
       " 5281: tensor([[-0.0206, -0.0297, -0.0015,  ..., -0.0116,  0.0012, -0.0049]]),\n",
       " 9658: tensor([[-0.0090, -0.0103,  0.0191,  ..., -0.0103, -0.0063, -0.0172]]),\n",
       " 5178: tensor([[ 0.0035, -0.0052, -0.0176,  ..., -0.0102, -0.0087, -0.0231]]),\n",
       " 10982: tensor([[-0.0098,  0.0054,  0.0329,  ..., -0.0167,  0.0031, -0.0093]]),\n",
       " 2725: tensor([[-1.9180e-02, -2.0919e-05,  1.2971e-02,  ..., -9.5235e-03,\n",
       "          -1.3788e-03, -1.9074e-02]]),\n",
       " 7520: tensor([[-1.7921e-02,  2.1717e-02,  8.6034e-03,  ...,  3.6107e-03,\n",
       "           6.3443e-05, -4.0736e-02]]),\n",
       " 1491: tensor([[-0.0247,  0.0127,  0.0004,  ..., -0.0111, -0.0295, -0.0122]]),\n",
       " 4694: tensor([[-0.0307, -0.0005,  0.0181,  ..., -0.0090, -0.0123, -0.0232]]),\n",
       " 1768: tensor([[-0.0287, -0.0050,  0.0257,  ..., -0.0284, -0.0158, -0.0178]]),\n",
       " 7442: tensor([[ 0.0013,  0.0169,  0.0277,  ..., -0.0021, -0.0026, -0.0258]]),\n",
       " 2647: tensor([[-0.0098,  0.0212,  0.0384,  ..., -0.0012, -0.0146, -0.0309]]),\n",
       " 4940: tensor([[-0.0264,  0.0040,  0.0151,  ..., -0.0174, -0.0324, -0.0110]]),\n",
       " 6135: tensor([[-0.0173,  0.0025,  0.0264,  ..., -0.0361, -0.0241, -0.0143]]),\n",
       " 3330: tensor([[-0.0210,  0.0155,  0.0196,  ..., -0.0025, -0.0278, -0.0004]]),\n",
       " 10734: tensor([[-0.0275,  0.0088,  0.0061,  ..., -0.0157, -0.0294, -0.0283]]),\n",
       " 2993: tensor([[-0.0125, -0.0139,  0.0130,  ..., -0.0329, -0.0329, -0.0058]]),\n",
       " 7217: tensor([[-0.0195,  0.0064,  0.0049,  ..., -0.0248, -0.0082, -0.0237]]),\n",
       " 834: tensor([[-0.0206,  0.0012, -0.0159,  ..., -0.0440, -0.0094,  0.0159]]),\n",
       " 2012: tensor([[-0.0061, -0.0081,  0.0087,  ..., -0.0300, -0.0150, -0.0059]]),\n",
       " 4338: tensor([[-0.0031,  0.0003,  0.0284,  ...,  0.0103, -0.0106, -0.0519]]),\n",
       " 8418: tensor([[ 0.0093,  0.0027,  0.0406,  ..., -0.0002,  0.0188, -0.0318]]),\n",
       " 3948: tensor([[-0.0334, -0.0012,  0.0088,  ...,  0.0071, -0.0312, -0.0164]]),\n",
       " 419: tensor([[ 0.0031, -0.0161,  0.0309,  ...,  0.0018, -0.0135, -0.0283]]),\n",
       " 9296: tensor([[-0.0044, -0.0032,  0.0249,  ..., -0.0220, -0.0145, -0.0367]]),\n",
       " 6760: tensor([[-0.0151,  0.0020,  0.0200,  ..., -0.0264, -0.0285, -0.0226]]),\n",
       " 8835: tensor([[-0.0046, -0.0155,  0.0277,  ..., -0.0102, -0.0104, -0.0338]]),\n",
       " 10298: tensor([[-0.0088,  0.0284,  0.0356,  ..., -0.0152, -0.0128, -0.0150]]),\n",
       " 10161: tensor([[-0.0040,  0.0072,  0.0214,  ..., -0.0060,  0.0073, -0.0237]]),\n",
       " 3565: tensor([[-0.0021,  0.0052,  0.0244,  ..., -0.0234, -0.0190, -0.0365]]),\n",
       " 6499: tensor([[-0.0150, -0.0101,  0.0396,  ..., -0.0084, -0.0387, -0.0307]]),\n",
       " 682: tensor([[ 0.0002, -0.0125,  0.0602,  ..., -0.0172, -0.0135, -0.0119]]),\n",
       " 3407: tensor([[ 0.0036, -0.0068,  0.0202,  ..., -0.0125, -0.0216, -0.0178]]),\n",
       " 10003: tensor([[ 0.0006, -0.0038,  0.0239,  ..., -0.0272, -0.0142, -0.0074]]),\n",
       " 8957: tensor([[-0.0084, -0.0109,  0.0050,  ..., -0.0084, -0.0098, -0.0119]]),\n",
       " 6602: tensor([[-0.0342,  0.0043,  0.0224,  ...,  0.0050,  0.0019, -0.0257]]),\n",
       " 2170: tensor([[ 0.0026,  0.0108,  0.0218,  ..., -0.0254,  0.0073, -0.0312]]),\n",
       " 956: tensor([[-0.0007,  0.0096,  0.0400,  ..., -0.0154, -0.0051, -0.0284]]),\n",
       " 2289: tensor([[-0.0406,  0.0026,  0.0101,  ..., -0.0200, -0.0009, -0.0021]]),\n",
       " 7375: tensor([[-0.0311, -0.0148,  0.0063,  ..., -0.0175, -0.0038, -0.0379]]),\n",
       " 8683: tensor([[-0.0135,  0.0045,  0.0242,  ..., -0.0253, -0.0360, -0.0121]]),\n",
       " 446: tensor([[-0.0353, -0.0076,  0.0028,  ..., -0.0172, -0.0231, -0.0101]]),\n",
       " 9130: tensor([[-0.0217, -0.0051,  0.0196,  ..., -0.0132, -0.0075, -0.0131]]),\n",
       " 5610: tensor([[-0.0212,  0.0164,  0.0059,  ..., -0.0088, -0.0092, -0.0246]]),\n",
       " 1162: tensor([[-0.0236, -0.0025, -0.0085,  ..., -0.0206, -0.0002, -0.0235]]),\n",
       " 3917: tensor([[-0.0479, -0.0060, -0.0083,  ..., -0.0090, -0.0081, -0.0250]]),\n",
       " 8447: tensor([[-0.0057, -0.0099,  0.0156,  ..., -0.0497, -0.0080, -0.0170]]),\n",
       " 4367: tensor([[-0.0178, -0.0264,  0.0218,  ...,  0.0028, -0.0128, -0.0255]]),\n",
       " 7248: tensor([[ 6.8541e-05,  1.6040e-02,  2.1287e-02,  ..., -1.6897e-02,\n",
       "          -2.7794e-02, -2.2118e-02]]),\n",
       " 8525: tensor([[-0.0341, -0.0132, -0.0105,  ..., -0.0139, -0.0316, -0.0103]]),\n",
       " 4205: tensor([[-0.0237,  0.0013,  0.0177,  ..., -0.0169, -0.0238, -0.0139]]),\n",
       " 6989: tensor([[-0.0096, -0.0202,  0.0148,  ..., -0.0068, -0.0085, -0.0273]]),\n",
       " 3875: tensor([[-0.0303, -0.0040,  0.0055,  ..., -0.0015, -0.0413, -0.0052]]),\n",
       " 1000: tensor([[ 0.0131,  0.0058,  0.0344,  ..., -0.0121, -0.0080, -0.0207]]),\n",
       " 909: tensor([[-0.0510, -0.0059, -0.0062,  ..., -0.0098,  0.0020, -0.0195]]),\n",
       " 8908: tensor([[-0.0144,  0.0086,  0.0311,  ..., -0.0130, -0.0227, -0.0249]]),\n",
       " 3458: tensor([[-0.0331, -0.0058, -0.0074,  ..., -0.0267, -0.0166, -0.0131]]),\n",
       " ...}"
      ]
     },
     "execution_count": 8,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Check query embeddings\n",
    "query_embeddings"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "(11204, torch.Size([1, 1536]))"
      ]
     },
     "execution_count": 14,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Check length and dimension of query embeddings\n",
    "len(query_embeddings), query_embeddings[0].shape"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{0: tensor([[-0.0497, -0.0080, -0.0108,  ..., -0.0098, -0.0167, -0.0184]]),\n",
       " 1: tensor([[-0.0310,  0.0156,  0.0047,  ..., -0.0299, -0.0185, -0.0211]]),\n",
       " 2: tensor([[-0.0295,  0.0115, -0.0203,  ..., -0.0397, -0.0181, -0.0204]]),\n",
       " 3: tensor([[-0.0320, -0.0132, -0.0218,  ..., -0.0329, -0.0123, -0.0272]]),\n",
       " 4: tensor([[-0.0325, -0.0123, -0.0177,  ..., -0.0282, -0.0045, -0.0315]]),\n",
       " 5: tensor([[-0.0136,  0.0236,  0.0044,  ..., -0.0408, -0.0187, -0.0272]]),\n",
       " 6: tensor([[-0.0257, -0.0240, -0.0326,  ..., -0.0291, -0.0035, -0.0589]]),\n",
       " 7: tensor([[-0.0396,  0.0123, -0.0340,  ..., -0.0194, -0.0149, -0.0235]]),\n",
       " 8: tensor([[-0.0345, -0.0122, -0.0214,  ..., -0.0268, -0.0206, -0.0376]]),\n",
       " 9: tensor([[-0.0239,  0.0046, -0.0159,  ..., -0.0240, -0.0084, -0.0385]]),\n",
       " 10: tensor([[-0.0468,  0.0205, -0.0175,  ..., -0.0329, -0.0131, -0.0329]]),\n",
       " 11: tensor([[-0.0305, -0.0085, -0.0157,  ..., -0.0219,  0.0019, -0.0292]]),\n",
       " 12: tensor([[-0.0392,  0.0038, -0.0234,  ..., -0.0421, -0.0152, -0.0192]]),\n",
       " 13: tensor([[-0.0346, -0.0029, -0.0270,  ..., -0.0328, -0.0126, -0.0392]]),\n",
       " 14: tensor([[-0.0262,  0.0027, -0.0106,  ..., -0.0319, -0.0069, -0.0253]]),\n",
       " 15: tensor([[-0.0490, -0.0068, -0.0229,  ..., -0.0331, -0.0051, -0.0285]]),\n",
       " 16: tensor([[-0.0375,  0.0165, -0.0160,  ..., -0.0340, -0.0232, -0.0056]]),\n",
       " 17: tensor([[-0.0311,  0.0126, -0.0076,  ..., -0.0138, -0.0285, -0.0084]]),\n",
       " 18: tensor([[-0.0543, -0.0118, -0.0111,  ..., -0.0184, -0.0166, -0.0349]]),\n",
       " 19: tensor([[-0.0484, -0.0021, -0.0089,  ..., -0.0289,  0.0047, -0.0100]]),\n",
       " 20: tensor([[-0.0246,  0.0050,  0.0015,  ..., -0.0201, -0.0234, -0.0226]]),\n",
       " 21: tensor([[-0.0323,  0.0020, -0.0073,  ..., -0.0208, -0.0021, -0.0086]]),\n",
       " 22: tensor([[-0.0352, -0.0054, -0.0212,  ..., -0.0306, -0.0104, -0.0211]]),\n",
       " 23: tensor([[-0.0442,  0.0124, -0.0185,  ..., -0.0354, -0.0137,  0.0015]]),\n",
       " 24: tensor([[-0.0395,  0.0162,  0.0140,  ..., -0.0434,  0.0019, -0.0188]]),\n",
       " 25: tensor([[-0.0507, -0.0074, -0.0236,  ..., -0.0168, -0.0180, -0.0305]]),\n",
       " 26: tensor([[-0.0464, -0.0071, -0.0220,  ..., -0.0388, -0.0096, -0.0215]]),\n",
       " 27: tensor([[-0.0241,  0.0031,  0.0005,  ..., -0.0172, -0.0211, -0.0140]]),\n",
       " 28: tensor([[-0.0398, -0.0016, -0.0021,  ..., -0.0274, -0.0072, -0.0229]]),\n",
       " 29: tensor([[-0.0483, -0.0168, -0.0316,  ..., -0.0265, -0.0036, -0.0249]]),\n",
       " 30: tensor([[-0.0516, -0.0005, -0.0175,  ..., -0.0351, -0.0135, -0.0192]]),\n",
       " 31: tensor([[-0.0338, -0.0124, -0.0109,  ..., -0.0212, -0.0079, -0.0219]]),\n",
       " 32: tensor([[-0.0310, -0.0007, -0.0193,  ..., -0.0381, -0.0324, -0.0135]]),\n",
       " 33: tensor([[-0.0378, -0.0063, -0.0011,  ..., -0.0309, -0.0144, -0.0240]]),\n",
       " 34: tensor([[-0.0246, -0.0071,  0.0018,  ..., -0.0319, -0.0094, -0.0082]]),\n",
       " 35: tensor([[-0.0119,  0.0198,  0.0092,  ..., -0.0352, -0.0150, -0.0320]]),\n",
       " 36: tensor([[-0.0344, -0.0056, -0.0079,  ..., -0.0477, -0.0266, -0.0092]]),\n",
       " 37: tensor([[-0.0206,  0.0138, -0.0174,  ..., -0.0389,  0.0007, -0.0240]]),\n",
       " 38: tensor([[-0.0453, -0.0030, -0.0072,  ..., -0.0313, -0.0186, -0.0271]]),\n",
       " 39: tensor([[-0.0560, -0.0014, -0.0270,  ..., -0.0159, -0.0054, -0.0269]]),\n",
       " 40: tensor([[-0.0377, -0.0008, -0.0217,  ..., -0.0237, -0.0286, -0.0138]]),\n",
       " 41: tensor([[-0.0371, -0.0117, -0.0135,  ..., -0.0497, -0.0031, -0.0318]]),\n",
       " 42: tensor([[-0.0335,  0.0034, -0.0234,  ..., -0.0025, -0.0142, -0.0506]]),\n",
       " 43: tensor([[-0.0301, -0.0157, -0.0153,  ..., -0.0235, -0.0070, -0.0193]]),\n",
       " 44: tensor([[-0.0201,  0.0050, -0.0203,  ..., -0.0426, -0.0196, -0.0168]]),\n",
       " 45: tensor([[-0.0470,  0.0032, -0.0271,  ..., -0.0089, -0.0067, -0.0090]]),\n",
       " 46: tensor([[-0.0334,  0.0033, -0.0152,  ..., -0.0250, -0.0189, -0.0190]]),\n",
       " 47: tensor([[-0.0354, -0.0274, -0.0133,  ..., -0.0273, -0.0133, -0.0255]]),\n",
       " 48: tensor([[-0.0353, -0.0092, -0.0234,  ..., -0.0227, -0.0118, -0.0284]]),\n",
       " 49: tensor([[-0.0344, -0.0043,  0.0104,  ..., -0.0149, -0.0034, -0.0386]]),\n",
       " 50: tensor([[-0.0475, -0.0132, -0.0267,  ..., -0.0219, -0.0075, -0.0262]]),\n",
       " 51: tensor([[-0.0374, -0.0038, -0.0065,  ..., -0.0216, -0.0189, -0.0263]]),\n",
       " 52: tensor([[-0.0389, -0.0061, -0.0157,  ..., -0.0278, -0.0094, -0.0173]]),\n",
       " 53: tensor([[-0.0520, -0.0026, -0.0457,  ..., -0.0336, -0.0097, -0.0281]]),\n",
       " 54: tensor([[-0.0280, -0.0075, -0.0208,  ..., -0.0251, -0.0181, -0.0239]]),\n",
       " 55: tensor([[-0.0433, -0.0098, -0.0194,  ..., -0.0352, -0.0144, -0.0173]]),\n",
       " 56: tensor([[-0.0516,  0.0055,  0.0027,  ..., -0.0162, -0.0002, -0.0205]]),\n",
       " 57: tensor([[-0.0297,  0.0080, -0.0182,  ..., -0.0204, -0.0075, -0.0241]]),\n",
       " 58: tensor([[-0.0365, -0.0027, -0.0091,  ..., -0.0243,  0.0030, -0.0126]]),\n",
       " 59: tensor([[-0.0384,  0.0015, -0.0135,  ..., -0.0195,  0.0002, -0.0232]]),\n",
       " 60: tensor([[-0.0345, -0.0034,  0.0086,  ..., -0.0230, -0.0242, -0.0116]]),\n",
       " 61: tensor([[-0.0311, -0.0028, -0.0196,  ..., -0.0178, -0.0059, -0.0364]]),\n",
       " 62: tensor([[-4.1921e-02,  9.5770e-03,  5.0823e-05,  ..., -1.6245e-02,\n",
       "          -1.2847e-04, -2.8425e-02]]),\n",
       " 63: tensor([[-0.0354,  0.0020,  0.0061,  ..., -0.0081, -0.0007, -0.0184]]),\n",
       " 64: tensor([[-0.0227,  0.0038, -0.0093,  ..., -0.0220, -0.0103, -0.0282]]),\n",
       " 65: tensor([[-0.0181,  0.0007, -0.0121,  ..., -0.0367, -0.0097, -0.0267]]),\n",
       " 66: tensor([[-0.0318,  0.0013, -0.0208,  ..., -0.0420, -0.0230, -0.0324]]),\n",
       " 67: tensor([[-0.0400, -0.0078, -0.0288,  ..., -0.0354, -0.0237, -0.0222]]),\n",
       " 68: tensor([[-0.0470, -0.0098, -0.0249,  ..., -0.0264, -0.0142, -0.0331]]),\n",
       " 69: tensor([[-0.0451, -0.0103, -0.0260,  ..., -0.0177, -0.0167, -0.0214]]),\n",
       " 70: tensor([[-0.0234,  0.0182, -0.0098,  ..., -0.0185, -0.0062, -0.0193]]),\n",
       " 71: tensor([[-0.0487, -0.0023, -0.0110,  ..., -0.0126, -0.0298, -0.0327]]),\n",
       " 72: tensor([[-0.0288, -0.0364, -0.0058,  ..., -0.0352,  0.0062, -0.0364]]),\n",
       " 73: tensor([[-0.0296, -0.0003, -0.0019,  ..., -0.0213, -0.0060, -0.0206]]),\n",
       " 74: tensor([[-0.0415,  0.0012, -0.0381,  ..., -0.0563, -0.0191, -0.0263]]),\n",
       " 75: tensor([[-0.0411,  0.0064, -0.0148,  ..., -0.0293, -0.0058, -0.0361]]),\n",
       " 76: tensor([[-0.0251, -0.0117, -0.0037,  ..., -0.0350, -0.0090, -0.0282]]),\n",
       " 77: tensor([[-0.0329, -0.0043, -0.0210,  ..., -0.0161,  0.0074, -0.0093]]),\n",
       " 78: tensor([[-0.0407,  0.0118, -0.0073,  ..., -0.0230, -0.0112, -0.0054]]),\n",
       " 79: tensor([[-0.0413, -0.0081, -0.0305,  ..., -0.0345, -0.0116, -0.0198]]),\n",
       " 80: tensor([[-0.0422,  0.0095, -0.0249,  ..., -0.0164, -0.0124, -0.0228]]),\n",
       " 81: tensor([[-0.0405, -0.0134, -0.0289,  ..., -0.0141,  0.0033, -0.0429]]),\n",
       " 82: tensor([[-0.0206,  0.0025, -0.0112,  ..., -0.0256, -0.0119, -0.0162]]),\n",
       " 83: tensor([[-0.0301,  0.0133, -0.0233,  ..., -0.0368, -0.0239, -0.0205]]),\n",
       " 84: tensor([[-0.0296, -0.0033, -0.0072,  ..., -0.0188, -0.0056, -0.0209]]),\n",
       " 85: tensor([[-0.0456, -0.0043, -0.0070,  ..., -0.0367, -0.0173, -0.0275]]),\n",
       " 86: tensor([[-0.0375,  0.0012, -0.0171,  ..., -0.0304, -0.0177, -0.0207]]),\n",
       " 87: tensor([[-0.0411,  0.0059, -0.0109,  ..., -0.0409, -0.0054, -0.0209]]),\n",
       " 88: tensor([[-0.0151,  0.0048, -0.0104,  ..., -0.0258, -0.0110, -0.0275]]),\n",
       " 89: tensor([[-2.8237e-02,  5.4525e-03, -8.6096e-03,  ..., -4.0161e-02,\n",
       "          -7.9270e-05, -3.0100e-02]]),\n",
       " 90: tensor([[-0.0369,  0.0002, -0.0167,  ..., -0.0555, -0.0223, -0.0213]]),\n",
       " 91: tensor([[-0.0425, -0.0039, -0.0130,  ..., -0.0251, -0.0104, -0.0252]]),\n",
       " 92: tensor([[-0.0307,  0.0134, -0.0030,  ..., -0.0342, -0.0055, -0.0333]]),\n",
       " 93: tensor([[-0.0530,  0.0143, -0.0053,  ..., -0.0317, -0.0006,  0.0060]]),\n",
       " 94: tensor([[-0.0431, -0.0255, -0.0246,  ..., -0.0265, -0.0033, -0.0318]]),\n",
       " 95: tensor([[-0.0226, -0.0106, -0.0131,  ..., -0.0502,  0.0056, -0.0303]]),\n",
       " 96: tensor([[-0.0274, -0.0168,  0.0043,  ..., -0.0344, -0.0127, -0.0207]]),\n",
       " 97: tensor([[-0.0317,  0.0062, -0.0047,  ..., -0.0330, -0.0030, -0.0188]]),\n",
       " 98: tensor([[-0.0274,  0.0116, -0.0038,  ..., -0.0246, -0.0041, -0.0244]]),\n",
       " 99: tensor([[-0.0239, -0.0111, -0.0183,  ..., -0.0484,  0.0027, -0.0192]]),\n",
       " 100: tensor([[-0.0245,  0.0184, -0.0053,  ..., -0.0245, -0.0022, -0.0226]]),\n",
       " 101: tensor([[-0.0299, -0.0202, -0.0104,  ..., -0.0205, -0.0163, -0.0373]]),\n",
       " 102: tensor([[-0.0498, -0.0018, -0.0117,  ..., -0.0277, -0.0127, -0.0313]]),\n",
       " 103: tensor([[-0.0366,  0.0001, -0.0224,  ..., -0.0290, -0.0062, -0.0176]]),\n",
       " 104: tensor([[-0.0192, -0.0029, -0.0123,  ..., -0.0387, -0.0109, -0.0266]]),\n",
       " 105: tensor([[-0.0374, -0.0104, -0.0039,  ..., -0.0125, -0.0114, -0.0287]]),\n",
       " 106: tensor([[-0.0376, -0.0001, -0.0204,  ..., -0.0255, -0.0039, -0.0355]]),\n",
       " 107: tensor([[-0.0209,  0.0090, -0.0321,  ..., -0.0253, -0.0107, -0.0296]]),\n",
       " 108: tensor([[-0.0338, -0.0023, -0.0233,  ..., -0.0204, -0.0026, -0.0293]]),\n",
       " 109: tensor([[-0.0525, -0.0051, -0.0316,  ..., -0.0283, -0.0244, -0.0108]]),\n",
       " 110: tensor([[-0.0289,  0.0050, -0.0027,  ..., -0.0214,  0.0006, -0.0238]]),\n",
       " 111: tensor([[-0.0490, -0.0038, -0.0013,  ..., -0.0080, -0.0197, -0.0370]]),\n",
       " 112: tensor([[-0.0183, -0.0045, -0.0177,  ..., -0.0198, -0.0150, -0.0136]]),\n",
       " 113: tensor([[-0.0491,  0.0082, -0.0178,  ..., -0.0344, -0.0128, -0.0137]]),\n",
       " 114: tensor([[-0.0242, -0.0039, -0.0085,  ..., -0.0305,  0.0001, -0.0308]]),\n",
       " 115: tensor([[-0.0473,  0.0098, -0.0209,  ..., -0.0385, -0.0060,  0.0005]]),\n",
       " 116: tensor([[-0.0503,  0.0008, -0.0157,  ..., -0.0241, -0.0052, -0.0231]]),\n",
       " 117: tensor([[-0.0347,  0.0126, -0.0129,  ..., -0.0230, -0.0046, -0.0059]]),\n",
       " 118: tensor([[-0.0188,  0.0216, -0.0105,  ..., -0.0439, -0.0268, -0.0163]]),\n",
       " 119: tensor([[-0.0478, -0.0132, -0.0228,  ..., -0.0187, -0.0102, -0.0321]]),\n",
       " 120: tensor([[-0.0529, -0.0066, -0.0264,  ..., -0.0242, -0.0209, -0.0216]]),\n",
       " 121: tensor([[-0.0246,  0.0045, -0.0094,  ..., -0.0243,  0.0052, -0.0250]]),\n",
       " 122: tensor([[-0.0274, -0.0037, -0.0143,  ..., -0.0245, -0.0243, -0.0157]]),\n",
       " 123: tensor([[-0.0335, -0.0108, -0.0298,  ..., -0.0279, -0.0118, -0.0259]]),\n",
       " 124: tensor([[-0.0196,  0.0027, -0.0198,  ..., -0.0335,  0.0015, -0.0435]]),\n",
       " 125: tensor([[-0.0393,  0.0012, -0.0273,  ..., -0.0385,  0.0002, -0.0242]]),\n",
       " 126: tensor([[-0.0312,  0.0124, -0.0205,  ..., -0.0191, -0.0026, -0.0168]]),\n",
       " 127: tensor([[-0.0430,  0.0038, -0.0316,  ..., -0.0452, -0.0011, -0.0220]]),\n",
       " 128: tensor([[-0.0327, -0.0120, -0.0240,  ..., -0.0379, -0.0080, -0.0292]]),\n",
       " 129: tensor([[-0.0405,  0.0157, -0.0155,  ..., -0.0177, -0.0103, -0.0098]]),\n",
       " 130: tensor([[-0.0476,  0.0034, -0.0139,  ..., -0.0320, -0.0095, -0.0329]]),\n",
       " 131: tensor([[-0.0472, -0.0043, -0.0129,  ..., -0.0290, -0.0104, -0.0375]]),\n",
       " 132: tensor([[-0.0137,  0.0057, -0.0113,  ..., -0.0282, -0.0202, -0.0321]]),\n",
       " 133: tensor([[-0.0238, -0.0075, -0.0035,  ..., -0.0341, -0.0190, -0.0189]]),\n",
       " 134: tensor([[-0.0357,  0.0039, -0.0110,  ..., -0.0259, -0.0066, -0.0266]]),\n",
       " 135: tensor([[-0.0224, -0.0073,  0.0037,  ..., -0.0113, -0.0081, -0.0307]]),\n",
       " 136: tensor([[-0.0258,  0.0034, -0.0109,  ..., -0.0342,  0.0011, -0.0084]]),\n",
       " 137: tensor([[-0.0300,  0.0064, -0.0174,  ..., -0.0408, -0.0082, -0.0053]]),\n",
       " 138: tensor([[-0.0295,  0.0151, -0.0228,  ..., -0.0358, -0.0034, -0.0165]]),\n",
       " 139: tensor([[-0.0382, -0.0018, -0.0107,  ..., -0.0239, -0.0095, -0.0149]]),\n",
       " 140: tensor([[-0.0267,  0.0130, -0.0235,  ..., -0.0288, -0.0187, -0.0247]]),\n",
       " 141: tensor([[-0.0381,  0.0173, -0.0138,  ..., -0.0413, -0.0053, -0.0329]]),\n",
       " 142: tensor([[-0.0426, -0.0167, -0.0220,  ..., -0.0205, -0.0123, -0.0297]]),\n",
       " 143: tensor([[-0.0089, -0.0094, -0.0130,  ..., -0.0268, -0.0099, -0.0166]]),\n",
       " 144: tensor([[-0.0409,  0.0018, -0.0288,  ..., -0.0156, -0.0080, -0.0203]]),\n",
       " 145: tensor([[-0.0411, -0.0061, -0.0046,  ..., -0.0249, -0.0034, -0.0405]]),\n",
       " 146: tensor([[-0.0370,  0.0163, -0.0327,  ..., -0.0237, -0.0093, -0.0406]]),\n",
       " 147: tensor([[-0.0448,  0.0056, -0.0336,  ..., -0.0350, -0.0028, -0.0370]]),\n",
       " 148: tensor([[-0.0260,  0.0217, -0.0152,  ..., -0.0260, -0.0124, -0.0323]]),\n",
       " 149: tensor([[-0.0300, -0.0059, -0.0325,  ..., -0.0393, -0.0074, -0.0125]]),\n",
       " 150: tensor([[-0.0329,  0.0052, -0.0277,  ..., -0.0308, -0.0140, -0.0195]]),\n",
       " 151: tensor([[-0.0582,  0.0107, -0.0032,  ..., -0.0350, -0.0104, -0.0242]]),\n",
       " 152: tensor([[-0.0395, -0.0047, -0.0253,  ..., -0.0418, -0.0147, -0.0186]]),\n",
       " 153: tensor([[-0.0496, -0.0098, -0.0172,  ..., -0.0190, -0.0151, -0.0340]]),\n",
       " 154: tensor([[-0.0336,  0.0064, -0.0434,  ..., -0.0241, -0.0145, -0.0406]]),\n",
       " 155: tensor([[-0.0430,  0.0100, -0.0247,  ..., -0.0126, -0.0219, -0.0305]]),\n",
       " 156: tensor([[-0.0299, -0.0091, -0.0241,  ..., -0.0382, -0.0306, -0.0007]]),\n",
       " 157: tensor([[-0.0337,  0.0032, -0.0129,  ..., -0.0313,  0.0037, -0.0298]]),\n",
       " 158: tensor([[-0.0368, -0.0129, -0.0151,  ..., -0.0413, -0.0179, -0.0297]]),\n",
       " 159: tensor([[-0.0389,  0.0041, -0.0204,  ..., -0.0288,  0.0102, -0.0092]]),\n",
       " 160: tensor([[-0.0384, -0.0044, -0.0048,  ..., -0.0264, -0.0224, -0.0304]]),\n",
       " 161: tensor([[-0.0233, -0.0057, -0.0151,  ..., -0.0197, -0.0084, -0.0319]]),\n",
       " 162: tensor([[-0.0203,  0.0294,  0.0080,  ..., -0.0460, -0.0255, -0.0181]]),\n",
       " 163: tensor([[-0.0270,  0.0183, -0.0161,  ..., -0.0312, -0.0123, -0.0278]]),\n",
       " 164: tensor([[-0.0351, -0.0145, -0.0119,  ..., -0.0344, -0.0047, -0.0290]]),\n",
       " 165: tensor([[-3.1572e-02, -6.5590e-05, -5.0487e-03,  ..., -2.2928e-02,\n",
       "          -2.6104e-02, -1.4848e-02]]),\n",
       " 166: tensor([[-0.0409, -0.0092, -0.0102,  ..., -0.0263, -0.0203, -0.0279]]),\n",
       " 167: tensor([[-0.0214,  0.0034, -0.0074,  ..., -0.0443, -0.0298, -0.0020]]),\n",
       " 168: tensor([[-0.0483, -0.0150, -0.0249,  ..., -0.0187, -0.0126, -0.0305]]),\n",
       " 169: tensor([[-0.0281,  0.0120, -0.0098,  ..., -0.0245, -0.0041, -0.0273]]),\n",
       " 170: tensor([[-0.0299,  0.0024, -0.0131,  ..., -0.0301, -0.0066, -0.0324]]),\n",
       " 171: tensor([[-0.0389, -0.0124, -0.0204,  ..., -0.0056, -0.0124, -0.0359]]),\n",
       " 172: tensor([[-0.0611,  0.0054, -0.0058,  ..., -0.0429, -0.0042, -0.0329]]),\n",
       " 173: tensor([[-0.0364, -0.0112, -0.0049,  ..., -0.0436, -0.0055, -0.0292]]),\n",
       " 174: tensor([[-0.0496, -0.0123, -0.0195,  ..., -0.0332, -0.0077, -0.0323]]),\n",
       " 175: tensor([[-0.0400,  0.0079, -0.0143,  ..., -0.0348, -0.0147, -0.0276]]),\n",
       " 176: tensor([[-0.0377, -0.0062, -0.0146,  ..., -0.0142, -0.0214, -0.0274]]),\n",
       " 177: tensor([[-0.0493, -0.0021,  0.0041,  ..., -0.0103, -0.0085, -0.0276]]),\n",
       " 178: tensor([[-0.0502, -0.0050, -0.0061,  ..., -0.0180, -0.0152, -0.0300]]),\n",
       " 179: tensor([[-0.0357, -0.0035, -0.0459,  ..., -0.0410, -0.0138, -0.0213]]),\n",
       " 180: tensor([[-0.0454, -0.0130, -0.0219,  ..., -0.0289, -0.0213, -0.0262]]),\n",
       " 181: tensor([[-0.0269, -0.0139, -0.0025,  ..., -0.0224, -0.0186, -0.0503]]),\n",
       " 182: tensor([[-0.0451,  0.0115, -0.0259,  ..., -0.0471, -0.0228, -0.0200]]),\n",
       " 183: tensor([[-0.0477, -0.0152, -0.0159,  ..., -0.0400,  0.0004, -0.0322]]),\n",
       " 184: tensor([[-0.0526, -0.0019, -0.0179,  ..., -0.0329, -0.0129, -0.0204]]),\n",
       " 185: tensor([[-0.0399, -0.0082, -0.0208,  ..., -0.0282, -0.0133, -0.0369]]),\n",
       " 186: tensor([[-0.0256,  0.0024, -0.0163,  ..., -0.0238, -0.0133, -0.0270]]),\n",
       " 187: tensor([[-0.0480, -0.0007, -0.0166,  ..., -0.0320,  0.0014, -0.0289]]),\n",
       " 188: tensor([[-0.0334, -0.0053, -0.0303,  ..., -0.0287, -0.0133, -0.0316]]),\n",
       " 189: tensor([[-0.0344, -0.0153, -0.0107,  ..., -0.0245,  0.0014, -0.0264]]),\n",
       " 190: tensor([[-0.0331,  0.0011, -0.0063,  ..., -0.0153, -0.0257, -0.0345]]),\n",
       " 191: tensor([[-0.0467,  0.0077, -0.0197,  ..., -0.0203, -0.0112, -0.0192]]),\n",
       " 192: tensor([[-0.0280, -0.0008, -0.0231,  ..., -0.0340,  0.0052, -0.0179]]),\n",
       " 193: tensor([[-0.0312,  0.0083, -0.0336,  ..., -0.0399,  0.0052, -0.0189]]),\n",
       " 194: tensor([[-0.0202, -0.0148, -0.0054,  ..., -0.0267, -0.0168, -0.0189]]),\n",
       " 195: tensor([[-0.0279, -0.0010,  0.0044,  ..., -0.0049, -0.0096, -0.0311]]),\n",
       " 196: tensor([[-0.0364,  0.0130, -0.0044,  ..., -0.0358,  0.0036,  0.0012]]),\n",
       " 197: tensor([[-0.0459, -0.0059, -0.0184,  ..., -0.0237, -0.0246, -0.0133]]),\n",
       " 198: tensor([[-0.0507, -0.0049, -0.0094,  ..., -0.0224, -0.0192, -0.0172]]),\n",
       " 199: tensor([[-0.0390, -0.0090, -0.0201,  ..., -0.0255,  0.0053, -0.0283]]),\n",
       " 200: tensor([[-0.0527, -0.0037, -0.0145,  ..., -0.0231, -0.0020, -0.0319]]),\n",
       " 201: tensor([[-0.0168,  0.0012, -0.0194,  ..., -0.0288, -0.0113, -0.0428]]),\n",
       " 202: tensor([[-0.0412,  0.0030, -0.0045,  ..., -0.0237, -0.0016, -0.0171]]),\n",
       " 203: tensor([[-0.0367,  0.0217, -0.0166,  ..., -0.0370, -0.0035, -0.0286]]),\n",
       " 204: tensor([[-0.0467, -0.0135, -0.0174,  ..., -0.0428, -0.0012, -0.0333]]),\n",
       " 205: tensor([[-0.0468, -0.0055, -0.0075,  ..., -0.0245, -0.0046, -0.0263]]),\n",
       " 206: tensor([[-0.0333, -0.0007, -0.0134,  ..., -0.0365, -0.0041, -0.0121]]),\n",
       " 207: tensor([[-0.0530,  0.0023, -0.0283,  ..., -0.0218, -0.0189, -0.0241]]),\n",
       " 208: tensor([[-0.0424, -0.0108, -0.0286,  ..., -0.0511,  0.0011, -0.0342]]),\n",
       " 209: tensor([[-0.0474, -0.0018, -0.0350,  ..., -0.0313, -0.0050, -0.0312]]),\n",
       " 210: tensor([[-0.0343, -0.0173, -0.0147,  ..., -0.0481, -0.0015, -0.0315]]),\n",
       " 211: tensor([[-0.0497,  0.0102, -0.0135,  ..., -0.0369, -0.0104, -0.0422]]),\n",
       " 212: tensor([[-0.0201,  0.0215, -0.0002,  ..., -0.0343, -0.0135, -0.0285]]),\n",
       " 213: tensor([[-0.0388,  0.0024, -0.0081,  ..., -0.0368, -0.0080, -0.0252]]),\n",
       " 214: tensor([[-0.0156,  0.0013, -0.0196,  ..., -0.0233, -0.0075, -0.0134]]),\n",
       " 215: tensor([[-0.0401,  0.0106, -0.0239,  ..., -0.0464, -0.0066, -0.0047]]),\n",
       " 216: tensor([[-0.0306, -0.0027,  0.0015,  ..., -0.0313, -0.0008, -0.0411]]),\n",
       " 217: tensor([[-0.0433, -0.0134, -0.0090,  ..., -0.0058, -0.0223, -0.0246]]),\n",
       " 218: tensor([[-0.0209,  0.0021, -0.0088,  ..., -0.0389, -0.0137, -0.0271]]),\n",
       " 219: tensor([[-0.0466,  0.0033, -0.0190,  ..., -0.0250, -0.0063, -0.0224]]),\n",
       " 220: tensor([[-0.0480, -0.0120, -0.0050,  ..., -0.0289,  0.0197, -0.0265]]),\n",
       " 221: tensor([[-3.5769e-02,  1.9257e-02, -1.9559e-02,  ..., -2.5228e-02,\n",
       "           1.7171e-05, -4.4636e-02]]),\n",
       " 222: tensor([[-0.0370,  0.0173, -0.0176,  ..., -0.0291, -0.0018,  0.0029]]),\n",
       " 223: tensor([[-0.0370,  0.0063, -0.0181,  ..., -0.0447, -0.0043, -0.0312]]),\n",
       " 224: tensor([[-0.0268,  0.0157, -0.0124,  ..., -0.0139, -0.0041, -0.0122]]),\n",
       " 225: tensor([[-0.0416, -0.0136, -0.0205,  ..., -0.0268, -0.0183, -0.0344]]),\n",
       " 226: tensor([[-0.0317, -0.0012, -0.0195,  ..., -0.0379, -0.0109, -0.0115]]),\n",
       " 227: tensor([[-0.0210,  0.0056, -0.0445,  ..., -0.0367, -0.0128, -0.0311]]),\n",
       " 228: tensor([[-0.0308,  0.0025, -0.0458,  ..., -0.0341, -0.0037, -0.0382]]),\n",
       " 229: tensor([[-0.0389, -0.0007, -0.0414,  ..., -0.0248, -0.0152, -0.0472]]),\n",
       " 230: tensor([[-0.0307, -0.0028, -0.0273,  ..., -0.0189, -0.0248, -0.0389]]),\n",
       " 231: tensor([[-0.0354, -0.0048, -0.0125,  ..., -0.0184,  0.0018, -0.0312]]),\n",
       " 232: tensor([[-0.0441, -0.0096, -0.0299,  ..., -0.0266, -0.0070, -0.0335]]),\n",
       " 233: tensor([[-0.0419,  0.0006,  0.0004,  ..., -0.0193, -0.0076, -0.0057]]),\n",
       " 234: tensor([[-0.0263,  0.0151, -0.0164,  ..., -0.0197, -0.0069, -0.0301]]),\n",
       " 235: tensor([[-0.0457, -0.0019, -0.0232,  ..., -0.0415, -0.0203, -0.0122]]),\n",
       " 236: tensor([[-0.0368, -0.0003, -0.0236,  ..., -0.0316, -0.0300, -0.0167]]),\n",
       " 237: tensor([[-0.0286, -0.0102, -0.0366,  ..., -0.0026, -0.0017, -0.0360]]),\n",
       " 238: tensor([[-0.0426, -0.0167, -0.0225,  ..., -0.0283, -0.0062, -0.0299]]),\n",
       " 239: tensor([[-4.5897e-02,  5.4448e-05, -7.8173e-03,  ..., -2.7477e-03,\n",
       "          -1.3819e-02, -3.5191e-02]]),\n",
       " 240: tensor([[-0.0246,  0.0125, -0.0041,  ..., -0.0223, -0.0139, -0.0196]]),\n",
       " 241: tensor([[-0.0450,  0.0133, -0.0142,  ..., -0.0147, -0.0051, -0.0152]]),\n",
       " 242: tensor([[-0.0289, -0.0045, -0.0081,  ..., -0.0259, -0.0092, -0.0214]]),\n",
       " 243: tensor([[-0.0156,  0.0030, -0.0195,  ..., -0.0320,  0.0054, -0.0179]]),\n",
       " 244: tensor([[-0.0296,  0.0033, -0.0179,  ..., -0.0443, -0.0117, -0.0141]]),\n",
       " 245: tensor([[-0.0273,  0.0126, -0.0011,  ..., -0.0223, -0.0178, -0.0295]]),\n",
       " 246: tensor([[-0.0374, -0.0039, -0.0002,  ..., -0.0327, -0.0060, -0.0226]]),\n",
       " 247: tensor([[-0.0338, -0.0075, -0.0116,  ..., -0.0138, -0.0142, -0.0377]]),\n",
       " 248: tensor([[-0.0444, -0.0058, -0.0069,  ..., -0.0499, -0.0144, -0.0239]]),\n",
       " 249: tensor([[-0.0160, -0.0018, -0.0053,  ..., -0.0515, -0.0071, -0.0249]]),\n",
       " 250: tensor([[-3.3330e-02,  1.6583e-02,  2.3919e-05,  ..., -3.6624e-02,\n",
       "          -1.7032e-02, -3.6216e-02]]),\n",
       " 251: tensor([[-0.0418, -0.0110, -0.0101,  ..., -0.0272, -0.0274, -0.0121]]),\n",
       " 252: tensor([[-0.0363,  0.0131, -0.0150,  ..., -0.0270, -0.0068, -0.0149]]),\n",
       " 253: tensor([[-0.0499,  0.0015, -0.0097,  ..., -0.0366, -0.0239, -0.0243]]),\n",
       " 254: tensor([[-0.0381, -0.0034, -0.0023,  ..., -0.0231, -0.0246, -0.0094]]),\n",
       " 255: tensor([[-0.0262, -0.0190,  0.0043,  ..., -0.0302, -0.0101, -0.0257]]),\n",
       " 256: tensor([[-0.0407,  0.0121, -0.0114,  ..., -0.0412, -0.0084, -0.0204]]),\n",
       " 257: tensor([[-0.0364,  0.0024, -0.0130,  ..., -0.0338, -0.0033, -0.0351]]),\n",
       " 258: tensor([[-0.0412, -0.0072, -0.0183,  ..., -0.0034, -0.0199, -0.0068]]),\n",
       " 259: tensor([[-0.0358,  0.0165, -0.0230,  ..., -0.0072, -0.0008, -0.0198]]),\n",
       " 260: tensor([[-0.0519,  0.0024,  0.0026,  ..., -0.0301, -0.0021, -0.0308]]),\n",
       " 261: tensor([[-0.0411,  0.0009, -0.0081,  ..., -0.0198, -0.0120, -0.0265]]),\n",
       " 262: tensor([[-0.0448, -0.0002, -0.0213,  ..., -0.0258, -0.0025, -0.0291]]),\n",
       " 263: tensor([[-0.0494, -0.0115, -0.0071,  ..., -0.0300, -0.0124, -0.0251]]),\n",
       " 264: tensor([[-0.0373,  0.0030, -0.0166,  ..., -0.0120, -0.0069, -0.0305]]),\n",
       " 265: tensor([[-0.0185,  0.0088, -0.0109,  ..., -0.0167, -0.0288, -0.0125]]),\n",
       " 266: tensor([[-0.0308, -0.0237, -0.0349,  ..., -0.0346, -0.0034, -0.0569]]),\n",
       " 267: tensor([[-0.0330, -0.0028, -0.0045,  ..., -0.0191, -0.0256, -0.0256]]),\n",
       " 268: tensor([[-0.0339, -0.0194, -0.0026,  ..., -0.0198, -0.0050, -0.0141]]),\n",
       " 269: tensor([[-0.0567, -0.0120, -0.0174,  ..., -0.0191, -0.0106, -0.0307]]),\n",
       " 270: tensor([[-0.0488, -0.0076, -0.0087,  ..., -0.0089, -0.0206, -0.0320]]),\n",
       " 271: tensor([[-4.3017e-02,  6.6914e-04, -4.6223e-02,  ..., -1.8947e-02,\n",
       "          -2.7908e-05, -5.2059e-02]]),\n",
       " 272: tensor([[-0.0352,  0.0165, -0.0002,  ..., -0.0245,  0.0030, -0.0328]]),\n",
       " 273: tensor([[-0.0411, -0.0162, -0.0094,  ..., -0.0362, -0.0122, -0.0145]]),\n",
       " 274: tensor([[-0.0226,  0.0036, -0.0135,  ..., -0.0288,  0.0046, -0.0150]]),\n",
       " 275: tensor([[-0.0444, -0.0020, -0.0185,  ..., -0.0237, -0.0070, -0.0238]]),\n",
       " 276: tensor([[-0.0339,  0.0142, -0.0244,  ..., -0.0340, -0.0051, -0.0345]]),\n",
       " 277: tensor([[-0.0263,  0.0184, -0.0093,  ..., -0.0251,  0.0124, -0.0296]]),\n",
       " 278: tensor([[-0.0282, -0.0055, -0.0073,  ..., -0.0220, -0.0181, -0.0222]]),\n",
       " 279: tensor([[-0.0268,  0.0003, -0.0224,  ..., -0.0303, -0.0002, -0.0247]]),\n",
       " 280: tensor([[-0.0303, -0.0222, -0.0142,  ..., -0.0406, -0.0010, -0.0224]]),\n",
       " 281: tensor([[-0.0287,  0.0053, -0.0262,  ..., -0.0243, -0.0161, -0.0362]]),\n",
       " 282: tensor([[-0.0433,  0.0155,  0.0047,  ..., -0.0391, -0.0131, -0.0212]]),\n",
       " 283: tensor([[-0.0298, -0.0079, -0.0095,  ..., -0.0240, -0.0195, -0.0111]]),\n",
       " 284: tensor([[-0.0367, -0.0004, -0.0203,  ..., -0.0404, -0.0075, -0.0157]]),\n",
       " 285: tensor([[-0.0405,  0.0050, -0.0153,  ..., -0.0313,  0.0060, -0.0275]]),\n",
       " 286: tensor([[-0.0437,  0.0132, -0.0241,  ..., -0.0322, -0.0126, -0.0334]]),\n",
       " 287: tensor([[-0.0318, -0.0131, -0.0245,  ..., -0.0237, -0.0249, -0.0252]]),\n",
       " 288: tensor([[-0.0384, -0.0041, -0.0306,  ..., -0.0384, -0.0213, -0.0401]]),\n",
       " 289: tensor([[-0.0205,  0.0057, -0.0161,  ..., -0.0409, -0.0109, -0.0295]]),\n",
       " 290: tensor([[-0.0390, -0.0073, -0.0082,  ..., -0.0256, -0.0029, -0.0194]]),\n",
       " 291: tensor([[-0.0413,  0.0023, -0.0162,  ..., -0.0244, -0.0112, -0.0205]]),\n",
       " 292: tensor([[-0.0312, -0.0006, -0.0125,  ..., -0.0389, -0.0248, -0.0205]]),\n",
       " 293: tensor([[-0.0475, -0.0093, -0.0215,  ..., -0.0174, -0.0356, -0.0223]]),\n",
       " 294: tensor([[-0.0279, -0.0049, -0.0206,  ..., -0.0241, -0.0047, -0.0243]]),\n",
       " 295: tensor([[-0.0441, -0.0087, -0.0090,  ..., -0.0019, -0.0066, -0.0355]]),\n",
       " 296: tensor([[-0.0479, -0.0002, -0.0158,  ..., -0.0171, -0.0082, -0.0237]]),\n",
       " 297: tensor([[-0.0311,  0.0077, -0.0198,  ..., -0.0368,  0.0031, -0.0226]]),\n",
       " 298: tensor([[-0.0334,  0.0084, -0.0128,  ..., -0.0015,  0.0051, -0.0205]]),\n",
       " 299: tensor([[-0.0211,  0.0057, -0.0072,  ..., -0.0234,  0.0029, -0.0224]]),\n",
       " 300: tensor([[-0.0342, -0.0056, -0.0017,  ..., -0.0291, -0.0223, -0.0287]]),\n",
       " 301: tensor([[-0.0234, -0.0045, -0.0113,  ..., -0.0256, -0.0189, -0.0292]]),\n",
       " 302: tensor([[-0.0349, -0.0016, -0.0114,  ..., -0.0335, -0.0199, -0.0398]]),\n",
       " 303: tensor([[-0.0405,  0.0053, -0.0068,  ..., -0.0338, -0.0044, -0.0322]]),\n",
       " 304: tensor([[-0.0345, -0.0017, -0.0169,  ..., -0.0213, -0.0026, -0.0294]]),\n",
       " 305: tensor([[-0.0331, -0.0126, -0.0055,  ..., -0.0107, -0.0089, -0.0227]]),\n",
       " 306: tensor([[-0.0462, -0.0074, -0.0201,  ..., -0.0078, -0.0123, -0.0076]]),\n",
       " 307: tensor([[-0.0308, -0.0191, -0.0110,  ..., -0.0462,  0.0028, -0.0179]]),\n",
       " 308: tensor([[-0.0294,  0.0057, -0.0127,  ..., -0.0289,  0.0091, -0.0224]]),\n",
       " 309: tensor([[-0.0447,  0.0061, -0.0152,  ..., -0.0213, -0.0138, -0.0301]]),\n",
       " 310: tensor([[-0.0335, -0.0065, -0.0030,  ..., -0.0242, -0.0174, -0.0337]]),\n",
       " 311: tensor([[-0.0170,  0.0120,  0.0033,  ..., -0.0182, -0.0189, -0.0326]]),\n",
       " 312: tensor([[-0.0385,  0.0074, -0.0182,  ..., -0.0343, -0.0099, -0.0193]]),\n",
       " 313: tensor([[-0.0489, -0.0138, -0.0164,  ..., -0.0282,  0.0054, -0.0134]]),\n",
       " 314: tensor([[-0.0399, -0.0097, -0.0244,  ..., -0.0313, -0.0244, -0.0269]]),\n",
       " 315: tensor([[-0.0412,  0.0002, -0.0271,  ..., -0.0369, -0.0158, -0.0155]]),\n",
       " 316: tensor([[-0.0372, -0.0054, -0.0062,  ..., -0.0189, -0.0238, -0.0199]]),\n",
       " 317: tensor([[-0.0241,  0.0047, -0.0093,  ..., -0.0266,  0.0096, -0.0338]]),\n",
       " 318: tensor([[-0.0294,  0.0066, -0.0009,  ..., -0.0249, -0.0065, -0.0283]]),\n",
       " 319: tensor([[-0.0255, -0.0010, -0.0072,  ..., -0.0238, -0.0073, -0.0185]]),\n",
       " 320: tensor([[-0.0386,  0.0154, -0.0198,  ..., -0.0342, -0.0111,  0.0112]]),\n",
       " 321: tensor([[-0.0376,  0.0027, -0.0374,  ..., -0.0125, -0.0032, -0.0333]]),\n",
       " 322: tensor([[-0.0330, -0.0116, -0.0253,  ..., -0.0278, -0.0045, -0.0317]]),\n",
       " 323: tensor([[-0.0248, -0.0041, -0.0189,  ..., -0.0119, -0.0078, -0.0459]]),\n",
       " 324: tensor([[-0.0275, -0.0091, -0.0145,  ..., -0.0388, -0.0136, -0.0194]]),\n",
       " 325: tensor([[-0.0317, -0.0172, -0.0139,  ..., -0.0308, -0.0069, -0.0322]]),\n",
       " 326: tensor([[-0.0451, -0.0062, -0.0128,  ..., -0.0371, -0.0091, -0.0145]]),\n",
       " 327: tensor([[-0.0325, -0.0019, -0.0157,  ..., -0.0274, -0.0075, -0.0236]]),\n",
       " 328: tensor([[-0.0329,  0.0081,  0.0067,  ..., -0.0107, -0.0218, -0.0095]]),\n",
       " 329: tensor([[-0.0367,  0.0080, -0.0039,  ..., -0.0132, -0.0076, -0.0084]]),\n",
       " 330: tensor([[-0.0312, -0.0051, -0.0063,  ..., -0.0087, -0.0041, -0.0269]]),\n",
       " 331: tensor([[-0.0421, -0.0103, -0.0320,  ..., -0.0067, -0.0087, -0.0169]]),\n",
       " 332: tensor([[-0.0415, -0.0050, -0.0343,  ..., -0.0307, -0.0235, -0.0191]]),\n",
       " 333: tensor([[-0.0246,  0.0025, -0.0334,  ..., -0.0358, -0.0131, -0.0454]]),\n",
       " 334: tensor([[-0.0498, -0.0128,  0.0125,  ..., -0.0297, -0.0227, -0.0438]]),\n",
       " 335: tensor([[-0.0482,  0.0045, -0.0296,  ..., -0.0311, -0.0162, -0.0268]]),\n",
       " 336: tensor([[-0.0551, -0.0054, -0.0050,  ..., -0.0281, -0.0239, -0.0200]]),\n",
       " 337: tensor([[-0.0511,  0.0026, -0.0252,  ..., -0.0401, -0.0227, -0.0224]]),\n",
       " 338: tensor([[-0.0411,  0.0167, -0.0129,  ..., -0.0189,  0.0027, -0.0185]]),\n",
       " 339: tensor([[-0.0303,  0.0091,  0.0117,  ..., -0.0081,  0.0032, -0.0269]]),\n",
       " 340: tensor([[-0.0422,  0.0069, -0.0086,  ..., -0.0381, -0.0162, -0.0210]]),\n",
       " 341: tensor([[-0.0313,  0.0058, -0.0124,  ..., -0.0229,  0.0034, -0.0351]]),\n",
       " 342: tensor([[-0.0379,  0.0010, -0.0280,  ..., -0.0298, -0.0180, -0.0142]]),\n",
       " 343: tensor([[-0.0375, -0.0150, -0.0199,  ..., -0.0086, -0.0103, -0.0322]]),\n",
       " 344: tensor([[-0.0369,  0.0126, -0.0220,  ..., -0.0372, -0.0116, -0.0241]]),\n",
       " 345: tensor([[-0.0309, -0.0153, -0.0111,  ..., -0.0389, -0.0014, -0.0200]]),\n",
       " 346: tensor([[-0.0346,  0.0067, -0.0079,  ..., -0.0303, -0.0045, -0.0162]]),\n",
       " 347: tensor([[-0.0407, -0.0003, -0.0216,  ..., -0.0288, -0.0024, -0.0082]]),\n",
       " 348: tensor([[-0.0378, -0.0040,  0.0169,  ..., -0.0002, -0.0143, -0.0360]]),\n",
       " 349: tensor([[-0.0324, -0.0066, -0.0169,  ..., -0.0179, -0.0022, -0.0268]]),\n",
       " 350: tensor([[-0.0527, -0.0028, -0.0162,  ..., -0.0431, -0.0144, -0.0341]]),\n",
       " 351: tensor([[-0.0499,  0.0090, -0.0164,  ..., -0.0255, -0.0125, -0.0253]]),\n",
       " 352: tensor([[-0.0316,  0.0079, -0.0141,  ..., -0.0329, -0.0040, -0.0265]]),\n",
       " 353: tensor([[-0.0268,  0.0156, -0.0138,  ..., -0.0117, -0.0300, -0.0123]]),\n",
       " 354: tensor([[-0.0278,  0.0094, -0.0360,  ..., -0.0463, -0.0104, -0.0143]]),\n",
       " 355: tensor([[-0.0349,  0.0154, -0.0300,  ..., -0.0267, -0.0127, -0.0178]]),\n",
       " 356: tensor([[-0.0475, -0.0164, -0.0167,  ..., -0.0422, -0.0184, -0.0274]]),\n",
       " 357: tensor([[-0.0367, -0.0101, -0.0195,  ..., -0.0410, -0.0167, -0.0205]]),\n",
       " 358: tensor([[-0.0640,  0.0053, -0.0264,  ..., -0.0179, -0.0295, -0.0390]]),\n",
       " 359: tensor([[-0.0366,  0.0030, -0.0155,  ..., -0.0226,  0.0049, -0.0091]]),\n",
       " 360: tensor([[-0.0500, -0.0115, -0.0340,  ..., -0.0275, -0.0121, -0.0219]]),\n",
       " 361: tensor([[-0.0237,  0.0075, -0.0102,  ..., -0.0284, -0.0205, -0.0207]]),\n",
       " 362: tensor([[-3.5714e-02, -2.5743e-03,  9.6236e-06,  ..., -2.2842e-02,\n",
       "          -7.7774e-03, -3.0538e-02]]),\n",
       " 363: tensor([[-0.0401,  0.0010, -0.0240,  ..., -0.0376, -0.0192, -0.0214]]),\n",
       " 364: tensor([[-0.0342,  0.0057, -0.0162,  ..., -0.0289, -0.0006, -0.0266]]),\n",
       " 365: tensor([[-0.0421,  0.0023, -0.0136,  ..., -0.0369, -0.0018,  0.0149]]),\n",
       " 366: tensor([[-0.0227,  0.0172, -0.0074,  ..., -0.0431, -0.0232, -0.0279]]),\n",
       " 367: tensor([[-0.0500, -0.0004, -0.0241,  ..., -0.0267, -0.0256, -0.0231]]),\n",
       " 368: tensor([[-0.0350, -0.0019, -0.0192,  ..., -0.0129, -0.0034, -0.0170]]),\n",
       " 369: tensor([[-0.0374,  0.0093, -0.0065,  ..., -0.0154, -0.0101, -0.0096]]),\n",
       " 370: tensor([[-0.0352,  0.0051, -0.0119,  ..., -0.0148, -0.0053, -0.0119]]),\n",
       " 371: tensor([[-0.0302, -0.0031, -0.0132,  ..., -0.0255, -0.0126, -0.0414]]),\n",
       " 372: tensor([[-0.0301, -0.0118, -0.0259,  ..., -0.0313, -0.0160, -0.0316]]),\n",
       " 373: tensor([[-0.0323,  0.0025, -0.0146,  ..., -0.0494, -0.0003, -0.0331]]),\n",
       " 374: tensor([[-0.0267, -0.0097, -0.0129,  ..., -0.0133, -0.0117, -0.0132]]),\n",
       " 375: tensor([[-0.0436, -0.0032, -0.0241,  ..., -0.0278, -0.0099, -0.0345]]),\n",
       " 376: tensor([[-0.0316, -0.0087, -0.0334,  ..., -0.0206, -0.0019, -0.0134]]),\n",
       " 377: tensor([[-0.0506,  0.0053, -0.0332,  ..., -0.0201, -0.0152, -0.0330]]),\n",
       " 378: tensor([[-0.0358, -0.0035, -0.0202,  ..., -0.0225, -0.0073, -0.0268]]),\n",
       " 379: tensor([[-0.0338, -0.0201, -0.0164,  ..., -0.0311, -0.0087, -0.0171]]),\n",
       " 380: tensor([[-0.0433,  0.0101, -0.0030,  ..., -0.0139, -0.0088, -0.0342]]),\n",
       " 381: tensor([[-0.0347,  0.0160, -0.0222,  ..., -0.0448,  0.0049, -0.0460]]),\n",
       " 382: tensor([[-0.0268, -0.0091, -0.0274,  ..., -0.0098, -0.0271, -0.0150]]),\n",
       " 383: tensor([[-0.0384, -0.0098, -0.0144,  ..., -0.0275,  0.0018, -0.0279]]),\n",
       " 384: tensor([[-0.0306,  0.0076, -0.0210,  ..., -0.0477, -0.0194, -0.0207]]),\n",
       " 385: tensor([[-0.0177, -0.0124, -0.0112,  ..., -0.0235, -0.0146, -0.0250]]),\n",
       " 386: tensor([[-0.0081,  0.0230, -0.0086,  ..., -0.0240, -0.0221, -0.0433]]),\n",
       " 387: tensor([[-0.0285, -0.0088, -0.0174,  ..., -0.0154, -0.0008, -0.0450]]),\n",
       " 388: tensor([[-0.0450, -0.0159, -0.0317,  ..., -0.0282, -0.0123, -0.0249]]),\n",
       " 389: tensor([[-0.0547, -0.0006, -0.0235,  ..., -0.0210, -0.0040, -0.0299]]),\n",
       " 390: tensor([[-0.0295,  0.0125, -0.0132,  ..., -0.0342, -0.0249, -0.0169]]),\n",
       " 391: tensor([[-0.0216, -0.0113, -0.0016,  ..., -0.0343, -0.0163, -0.0379]]),\n",
       " 392: tensor([[-0.0524, -0.0101, -0.0411,  ..., -0.0159, -0.0166, -0.0338]]),\n",
       " 393: tensor([[-0.0392, -0.0028, -0.0138,  ..., -0.0393, -0.0170, -0.0321]]),\n",
       " 394: tensor([[-0.0284,  0.0108, -0.0138,  ..., -0.0269, -0.0070, -0.0170]]),\n",
       " 395: tensor([[-0.0504,  0.0023, -0.0367,  ..., -0.0172, -0.0084, -0.0246]]),\n",
       " 396: tensor([[-0.0177,  0.0210,  0.0063,  ..., -0.0303, -0.0188, -0.0211]]),\n",
       " 397: tensor([[-0.0422, -0.0085, -0.0170,  ..., -0.0287, -0.0193, -0.0229]]),\n",
       " 398: tensor([[-0.0250, -0.0042,  0.0044,  ..., -0.0141, -0.0214, -0.0093]]),\n",
       " 399: tensor([[-0.0363, -0.0004, -0.0043,  ..., -0.0204, -0.0198, -0.0272]]),\n",
       " 400: tensor([[-0.0343,  0.0040, -0.0175,  ..., -0.0391, -0.0142, -0.0227]]),\n",
       " 401: tensor([[-0.0278,  0.0080, -0.0082,  ..., -0.0333, -0.0024, -0.0199]]),\n",
       " 402: tensor([[-0.0431, -0.0013, -0.0317,  ..., -0.0309, -0.0202, -0.0264]]),\n",
       " 403: tensor([[-0.0320, -0.0091, -0.0081,  ..., -0.0310, -0.0017, -0.0318]]),\n",
       " 404: tensor([[-0.0324,  0.0077, -0.0010,  ..., -0.0268, -0.0156, -0.0262]]),\n",
       " 405: tensor([[-0.0481, -0.0032, -0.0335,  ..., -0.0532,  0.0008, -0.0261]]),\n",
       " 406: tensor([[-0.0367, -0.0061, -0.0235,  ..., -0.0283, -0.0079, -0.0326]]),\n",
       " 407: tensor([[-0.0375, -0.0031, -0.0301,  ..., -0.0289, -0.0227, -0.0260]]),\n",
       " 408: tensor([[-0.0418, -0.0145, -0.0239,  ..., -0.0117, -0.0108, -0.0272]]),\n",
       " 409: tensor([[-0.0266, -0.0065, -0.0210,  ..., -0.0141, -0.0076, -0.0326]]),\n",
       " 410: tensor([[-0.0471,  0.0004, -0.0159,  ..., -0.0346, -0.0356, -0.0289]]),\n",
       " 411: tensor([[-0.0420, -0.0085, -0.0202,  ..., -0.0148, -0.0173, -0.0013]]),\n",
       " 412: tensor([[-0.0258,  0.0058, -0.0054,  ..., -0.0445, -0.0101, -0.0155]]),\n",
       " 413: tensor([[-0.0237, -0.0100,  0.0055,  ..., -0.0190, -0.0230, -0.0181]]),\n",
       " 414: tensor([[-0.0244,  0.0124, -0.0154,  ..., -0.0162, -0.0101, -0.0327]]),\n",
       " 415: tensor([[-0.0385, -0.0112, -0.0232,  ..., -0.0213, -0.0202, -0.0064]]),\n",
       " 416: tensor([[-0.0600, -0.0096, -0.0190,  ..., -0.0336, -0.0036, -0.0328]]),\n",
       " 417: tensor([[-0.0354,  0.0007, -0.0054,  ..., -0.0427, -0.0243, -0.0190]]),\n",
       " 418: tensor([[-0.0396, -0.0130, -0.0277,  ..., -0.0261, -0.0134, -0.0145]]),\n",
       " 419: tensor([[-0.0264,  0.0091,  0.0099,  ..., -0.0246, -0.0048, -0.0272]]),\n",
       " 420: tensor([[-0.0345, -0.0032, -0.0164,  ..., -0.0079, -0.0217, -0.0207]]),\n",
       " 421: tensor([[-0.0364, -0.0018, -0.0249,  ..., -0.0271, -0.0115, -0.0251]]),\n",
       " 422: tensor([[-0.0528, -0.0182, -0.0187,  ..., -0.0235, -0.0037, -0.0157]]),\n",
       " 423: tensor([[-0.0323, -0.0105, -0.0131,  ..., -0.0312, -0.0184, -0.0411]]),\n",
       " 424: tensor([[-0.0574,  0.0074, -0.0218,  ..., -0.0254, -0.0270, -0.0271]]),\n",
       " 425: tensor([[-0.0489,  0.0013,  0.0005,  ..., -0.0359, -0.0133, -0.0295]]),\n",
       " 426: tensor([[-0.0426,  0.0149, -0.0183,  ..., -0.0302, -0.0051, -0.0160]]),\n",
       " 427: tensor([[-0.0437,  0.0176, -0.0252,  ..., -0.0173,  0.0003, -0.0349]]),\n",
       " 428: tensor([[-0.0464,  0.0026, -0.0200,  ..., -0.0329,  0.0029, -0.0408]]),\n",
       " 429: tensor([[-0.0468, -0.0090, -0.0098,  ..., -0.0350, -0.0013, -0.0287]]),\n",
       " 430: tensor([[-0.0396,  0.0003, -0.0153,  ..., -0.0302, -0.0159, -0.0213]]),\n",
       " 431: tensor([[-0.0350,  0.0051, -0.0126,  ..., -0.0256, -0.0154, -0.0183]]),\n",
       " 432: tensor([[-0.0382, -0.0029, -0.0119,  ..., -0.0221, -0.0170, -0.0350]]),\n",
       " 433: tensor([[-0.0324, -0.0059, -0.0305,  ..., -0.0128, -0.0200, -0.0114]]),\n",
       " 434: tensor([[-0.0590, -0.0112, -0.0329,  ..., -0.0380,  0.0003, -0.0290]]),\n",
       " 435: tensor([[-3.9294e-02, -3.1003e-05, -9.5785e-03,  ..., -2.3539e-02,\n",
       "          -1.9567e-02, -3.2198e-02]]),\n",
       " 436: tensor([[-0.0411, -0.0115, -0.0013,  ..., -0.0470, -0.0108, -0.0146]]),\n",
       " 437: tensor([[-0.0143, -0.0217, -0.0124,  ..., -0.0168, -0.0140, -0.0099]]),\n",
       " 438: tensor([[-0.0467, -0.0130, -0.0088,  ..., -0.0412, -0.0089, -0.0291]]),\n",
       " 439: tensor([[-0.0164,  0.0025,  0.0060,  ..., -0.0200, -0.0090, -0.0258]]),\n",
       " 440: tensor([[-0.0390,  0.0082, -0.0120,  ..., -0.0680, -0.0273, -0.0081]]),\n",
       " 441: tensor([[-0.0400,  0.0068, -0.0335,  ..., -0.0382, -0.0133, -0.0174]]),\n",
       " 442: tensor([[-0.0231, -0.0136, -0.0187,  ..., -0.0186, -0.0203, -0.0225]]),\n",
       " 443: tensor([[-0.0500, -0.0140, -0.0191,  ..., -0.0176, -0.0032, -0.0377]]),\n",
       " 444: tensor([[-0.0406, -0.0089, -0.0352,  ..., -0.0090, -0.0113, -0.0274]]),\n",
       " 445: tensor([[-0.0393, -0.0064, -0.0188,  ..., -0.0334, -0.0076, -0.0131]]),\n",
       " 446: tensor([[-0.0356, -0.0019, -0.0136,  ..., -0.0234, -0.0115, -0.0310]]),\n",
       " 447: tensor([[-0.0365,  0.0087, -0.0143,  ..., -0.0341, -0.0095, -0.0255]]),\n",
       " 448: tensor([[-0.0455, -0.0014, -0.0170,  ..., -0.0287, -0.0204, -0.0314]]),\n",
       " 449: tensor([[-0.0476, -0.0074, -0.0290,  ..., -0.0273, -0.0155, -0.0436]]),\n",
       " 450: tensor([[-0.0325,  0.0131,  0.0040,  ..., -0.0112, -0.0039, -0.0304]]),\n",
       " 451: tensor([[-0.0240, -0.0027, -0.0118,  ..., -0.0312, -0.0097, -0.0166]]),\n",
       " 452: tensor([[-0.0289,  0.0025, -0.0002,  ..., -0.0192, -0.0139, -0.0349]]),\n",
       " 453: tensor([[-0.0397,  0.0047, -0.0120,  ..., -0.0368, -0.0026, -0.0085]]),\n",
       " 454: tensor([[-0.0581,  0.0097, -0.0182,  ..., -0.0268, -0.0220, -0.0185]]),\n",
       " 455: tensor([[-0.0245,  0.0149, -0.0129,  ..., -0.0257, -0.0122, -0.0388]]),\n",
       " 456: tensor([[-0.0367, -0.0081, -0.0336,  ..., -0.0336, -0.0014, -0.0285]]),\n",
       " 457: tensor([[-0.0437,  0.0134, -0.0134,  ..., -0.0266, -0.0267, -0.0156]]),\n",
       " 458: tensor([[-0.0344, -0.0098, -0.0251,  ..., -0.0362,  0.0033, -0.0233]]),\n",
       " 459: tensor([[-0.0398, -0.0121, -0.0044,  ..., -0.0138, -0.0287, -0.0174]]),\n",
       " 460: tensor([[-0.0391,  0.0067, -0.0132,  ..., -0.0280, -0.0251, -0.0249]]),\n",
       " 461: tensor([[-0.0091, -0.0004,  0.0034,  ..., -0.0272, -0.0093, -0.0210]]),\n",
       " 462: tensor([[-0.0319, -0.0192, -0.0251,  ..., -0.0287, -0.0136, -0.0130]]),\n",
       " 463: tensor([[-0.0269, -0.0067, -0.0110,  ..., -0.0396, -0.0197, -0.0220]]),\n",
       " 464: tensor([[-0.0380, -0.0146, -0.0376,  ..., -0.0293, -0.0030, -0.0437]]),\n",
       " 465: tensor([[-0.0387,  0.0131, -0.0111,  ..., -0.0381,  0.0064, -0.0219]]),\n",
       " 466: tensor([[-0.0251,  0.0075, -0.0273,  ..., -0.0498, -0.0125, -0.0200]]),\n",
       " 467: tensor([[-0.0576, -0.0008, -0.0145,  ..., -0.0436, -0.0216, -0.0231]]),\n",
       " 468: tensor([[-0.0521,  0.0033, -0.0111,  ..., -0.0239, -0.0021, -0.0108]]),\n",
       " 469: tensor([[-0.0489,  0.0003, -0.0240,  ..., -0.0372,  0.0038, -0.0034]]),\n",
       " 470: tensor([[-0.0437, -0.0092, -0.0247,  ..., -0.0221, -0.0277, -0.0314]]),\n",
       " 471: tensor([[-0.0420, -0.0052, -0.0173,  ..., -0.0279, -0.0149, -0.0252]]),\n",
       " 472: tensor([[-0.0296, -0.0046, -0.0220,  ..., -0.0333, -0.0173, -0.0348]]),\n",
       " 473: tensor([[-0.0440, -0.0020, -0.0155,  ..., -0.0220, -0.0191, -0.0303]]),\n",
       " 474: tensor([[-0.0196, -0.0036, -0.0313,  ..., -0.0378, -0.0309, -0.0205]]),\n",
       " 475: tensor([[-0.0129, -0.0014, -0.0128,  ..., -0.0360, -0.0211, -0.0216]]),\n",
       " 476: tensor([[-0.0306, -0.0210, -0.0205,  ..., -0.0286, -0.0047, -0.0397]]),\n",
       " 477: tensor([[-0.0304,  0.0153, -0.0301,  ..., -0.0577, -0.0104, -0.0226]]),\n",
       " 478: tensor([[-0.0381,  0.0143, -0.0151,  ..., -0.0195,  0.0015, -0.0222]]),\n",
       " 479: tensor([[-0.0342, -0.0147, -0.0061,  ..., -0.0217, -0.0075, -0.0455]]),\n",
       " 480: tensor([[-0.0263,  0.0103, -0.0021,  ..., -0.0320, -0.0101, -0.0247]]),\n",
       " 481: tensor([[-0.0243,  0.0128,  0.0010,  ..., -0.0285, -0.0159, -0.0211]]),\n",
       " 482: tensor([[-0.0449,  0.0002, -0.0106,  ..., -0.0213, -0.0146, -0.0336]]),\n",
       " 483: tensor([[-0.0321, -0.0155, -0.0312,  ..., -0.0272, -0.0125, -0.0257]]),\n",
       " 484: tensor([[-0.0335, -0.0086, -0.0340,  ..., -0.0185, -0.0222, -0.0213]]),\n",
       " 485: tensor([[-0.0340,  0.0163, -0.0121,  ..., -0.0247, -0.0257, -0.0322]]),\n",
       " 486: tensor([[-0.0162, -0.0007, -0.0314,  ..., -0.0159, -0.0143, -0.0380]]),\n",
       " 487: tensor([[-0.0291,  0.0069,  0.0076,  ..., -0.0442, -0.0261, -0.0225]]),\n",
       " 488: tensor([[-0.0286, -0.0032, -0.0156,  ..., -0.0379,  0.0007, -0.0276]]),\n",
       " 489: tensor([[-0.0423,  0.0028, -0.0082,  ..., -0.0242, -0.0112, -0.0185]]),\n",
       " 490: tensor([[-0.0421,  0.0006, -0.0250,  ..., -0.0409, -0.0216, -0.0194]]),\n",
       " 491: tensor([[-0.0355, -0.0105, -0.0196,  ..., -0.0400, -0.0046, -0.0421]]),\n",
       " 492: tensor([[-0.0376,  0.0005, -0.0238,  ..., -0.0157, -0.0257, -0.0198]]),\n",
       " 493: tensor([[-0.0346,  0.0110, -0.0029,  ..., -0.0223, -0.0142, -0.0228]]),\n",
       " 494: tensor([[-0.0117, -0.0241, -0.0314,  ..., -0.0142, -0.0066, -0.0262]]),\n",
       " 495: tensor([[-0.0452,  0.0069, -0.0119,  ..., -0.0182, -0.0106, -0.0132]]),\n",
       " 496: tensor([[-0.0278, -0.0083, -0.0115,  ..., -0.0291, -0.0184, -0.0254]]),\n",
       " 497: tensor([[-0.0229, -0.0114, -0.0301,  ..., -0.0367, -0.0096, -0.0218]]),\n",
       " 498: tensor([[-0.0339, -0.0159, -0.0228,  ..., -0.0148, -0.0216, -0.0398]]),\n",
       " 499: tensor([[-0.0363,  0.0056, -0.0262,  ..., -0.0287, -0.0093, -0.0342]]),\n",
       " 500: tensor([[-0.0254,  0.0155, -0.0184,  ..., -0.0402,  0.0026, -0.0279]]),\n",
       " 501: tensor([[-0.0362,  0.0085, -0.0059,  ..., -0.0142, -0.0217, -0.0323]]),\n",
       " 502: tensor([[-0.0430, -0.0055, -0.0092,  ..., -0.0154,  0.0026, -0.0238]]),\n",
       " 503: tensor([[-0.0553, -0.0086, -0.0203,  ..., -0.0362, -0.0193, -0.0324]]),\n",
       " 504: tensor([[-0.0282,  0.0097, -0.0133,  ..., -0.0125, -0.0029, -0.0295]]),\n",
       " 505: tensor([[-0.0300,  0.0023, -0.0108,  ..., -0.0273, -0.0006, -0.0425]]),\n",
       " 506: tensor([[-0.0455, -0.0019, -0.0285,  ..., -0.0342, -0.0107, -0.0323]]),\n",
       " 507: tensor([[-0.0299,  0.0058, -0.0267,  ..., -0.0225, -0.0045, -0.0219]]),\n",
       " 508: tensor([[-0.0237,  0.0024, -0.0191,  ..., -0.0190, -0.0068, -0.0256]]),\n",
       " 509: tensor([[-0.0521, -0.0034, -0.0055,  ..., -0.0296, -0.0115, -0.0252]]),\n",
       " 510: tensor([[-0.0572,  0.0039, -0.0258,  ..., -0.0446, -0.0185, -0.0009]]),\n",
       " 511: tensor([[-0.0503,  0.0048, -0.0100,  ..., -0.0342, -0.0255, -0.0210]]),\n",
       " 512: tensor([[-0.0348,  0.0150, -0.0144,  ..., -0.0033, -0.0069, -0.0319]]),\n",
       " 513: tensor([[-0.0390,  0.0144, -0.0067,  ..., -0.0385, -0.0270, -0.0188]]),\n",
       " 514: tensor([[-0.0348, -0.0070, -0.0123,  ..., -0.0357, -0.0075, -0.0172]]),\n",
       " 515: tensor([[-0.0429, -0.0117, -0.0063,  ..., -0.0321, -0.0134, -0.0255]]),\n",
       " 516: tensor([[-0.0524, -0.0003, -0.0021,  ..., -0.0183, -0.0056, -0.0311]]),\n",
       " 517: tensor([[-0.0287,  0.0050, -0.0160,  ..., -0.0311,  0.0064, -0.0339]]),\n",
       " 518: tensor([[-1.9234e-02, -2.4857e-05, -1.7192e-02,  ..., -2.1783e-02,\n",
       "          -9.6503e-03, -3.9509e-02]]),\n",
       " 519: tensor([[-0.0345,  0.0040, -0.0151,  ..., -0.0321, -0.0068, -0.0212]]),\n",
       " 520: tensor([[-0.0406,  0.0027, -0.0060,  ..., -0.0251,  0.0038, -0.0088]]),\n",
       " 521: tensor([[-0.0371, -0.0024, -0.0164,  ..., -0.0225, -0.0065, -0.0047]]),\n",
       " 522: tensor([[-0.0306, -0.0071, -0.0101,  ..., -0.0315, -0.0399, -0.0230]]),\n",
       " 523: tensor([[-0.0361, -0.0013, -0.0061,  ..., -0.0177, -0.0133, -0.0234]]),\n",
       " 524: tensor([[-0.0274, -0.0086, -0.0234,  ..., -0.0256, -0.0029, -0.0348]]),\n",
       " 525: tensor([[-0.0337, -0.0026, -0.0160,  ..., -0.0205, -0.0050, -0.0270]]),\n",
       " 526: tensor([[-0.0274,  0.0044, -0.0221,  ..., -0.0527, -0.0135, -0.0320]]),\n",
       " 527: tensor([[-0.0320,  0.0009, -0.0085,  ..., -0.0305,  0.0063, -0.0109]]),\n",
       " 528: tensor([[-0.0567,  0.0053, -0.0219,  ..., -0.0235, -0.0041, -0.0057]]),\n",
       " 529: tensor([[-0.0302,  0.0047, -0.0082,  ..., -0.0333, -0.0058, -0.0236]]),\n",
       " 530: tensor([[-3.7443e-02,  8.4775e-05, -2.7732e-02,  ..., -3.0618e-02,\n",
       "          -2.1500e-02, -2.4953e-02]]),\n",
       " 531: tensor([[-0.0386, -0.0132, -0.0159,  ..., -0.0174, -0.0057, -0.0293]]),\n",
       " 532: tensor([[-0.0201, -0.0064, -0.0234,  ..., -0.0163,  0.0004, -0.0245]]),\n",
       " 533: tensor([[-0.0406, -0.0169, -0.0412,  ..., -0.0305, -0.0087, -0.0368]]),\n",
       " 534: tensor([[-0.0426, -0.0123, -0.0087,  ..., -0.0352, -0.0152, -0.0339]]),\n",
       " 535: tensor([[-0.0270,  0.0024, -0.0144,  ..., -0.0357, -0.0207, -0.0104]]),\n",
       " 536: tensor([[-0.0424,  0.0033, -0.0179,  ..., -0.0268, -0.0202, -0.0191]]),\n",
       " 537: tensor([[-0.0187, -0.0002,  0.0019,  ..., -0.0233, -0.0169, -0.0029]]),\n",
       " 538: tensor([[-0.0423, -0.0016, -0.0060,  ..., -0.0270, -0.0057, -0.0102]]),\n",
       " 539: tensor([[-0.0252,  0.0115, -0.0044,  ..., -0.0265, -0.0090, -0.0333]]),\n",
       " 540: tensor([[-0.0127,  0.0071, -0.0051,  ..., -0.0303, -0.0267, -0.0221]]),\n",
       " 541: tensor([[-0.0446,  0.0139, -0.0052,  ..., -0.0245, -0.0014, -0.0380]]),\n",
       " 542: tensor([[-0.0227,  0.0029, -0.0318,  ..., -0.0521, -0.0103, -0.0200]]),\n",
       " 543: tensor([[-0.0476, -0.0127, -0.0295,  ..., -0.0325, -0.0082, -0.0230]]),\n",
       " 544: tensor([[-0.0328, -0.0037, -0.0350,  ..., -0.0125,  0.0088, -0.0297]]),\n",
       " 545: tensor([[-0.0486, -0.0042, -0.0114,  ..., -0.0304, -0.0036, -0.0326]]),\n",
       " 546: tensor([[-0.0363, -0.0005, -0.0204,  ..., -0.0324, -0.0219, -0.0255]]),\n",
       " 547: tensor([[-0.0424,  0.0057, -0.0165,  ..., -0.0488, -0.0070, -0.0273]]),\n",
       " 548: tensor([[-0.0520, -0.0023, -0.0066,  ..., -0.0374, -0.0160, -0.0231]]),\n",
       " 549: tensor([[-0.0407, -0.0090, -0.0260,  ..., -0.0343, -0.0110, -0.0333]]),\n",
       " 550: tensor([[-0.0195, -0.0183, -0.0184,  ..., -0.0180, -0.0142, -0.0247]]),\n",
       " 551: tensor([[-0.0306, -0.0020, -0.0121,  ..., -0.0190, -0.0004, -0.0352]]),\n",
       " 552: tensor([[-0.0219, -0.0074,  0.0018,  ..., -0.0276, -0.0094, -0.0262]]),\n",
       " 553: tensor([[-0.0277,  0.0022, -0.0200,  ..., -0.0286, -0.0123, -0.0218]]),\n",
       " 554: tensor([[-0.0394,  0.0040, -0.0189,  ..., -0.0159, -0.0133, -0.0258]]),\n",
       " 555: tensor([[-0.0416,  0.0009, -0.0160,  ..., -0.0189,  0.0012, -0.0128]]),\n",
       " 556: tensor([[-0.0476,  0.0007, -0.0060,  ..., -0.0104, -0.0201, -0.0299]]),\n",
       " 557: tensor([[-0.0526, -0.0006, -0.0199,  ..., -0.0436, -0.0205, -0.0224]]),\n",
       " 558: tensor([[-0.0456, -0.0079, -0.0086,  ..., -0.0427, -0.0141, -0.0198]]),\n",
       " 559: tensor([[-0.0323, -0.0013, -0.0133,  ..., -0.0126, -0.0034, -0.0221]]),\n",
       " 560: tensor([[-0.0283,  0.0161, -0.0157,  ..., -0.0225, -0.0088, -0.0363]]),\n",
       " 561: tensor([[-0.0377,  0.0048, -0.0174,  ..., -0.0469, -0.0139, -0.0120]]),\n",
       " 562: tensor([[-0.0423,  0.0053, -0.0321,  ..., -0.0285, -0.0084, -0.0270]]),\n",
       " 563: tensor([[-0.0362,  0.0043, -0.0208,  ..., -0.0102, -0.0175, -0.0379]]),\n",
       " 564: tensor([[-0.0285,  0.0080, -0.0197,  ..., -0.0227, -0.0075, -0.0116]]),\n",
       " 565: tensor([[-0.0349, -0.0025, -0.0224,  ..., -0.0164, -0.0111, -0.0306]]),\n",
       " 566: tensor([[-0.0085, -0.0063, -0.0071,  ..., -0.0224, -0.0181, -0.0211]]),\n",
       " 567: tensor([[-0.0289, -0.0110, -0.0116,  ..., -0.0329,  0.0019, -0.0299]]),\n",
       " 568: tensor([[-0.0402,  0.0008, -0.0232,  ..., -0.0219, -0.0178, -0.0237]]),\n",
       " 569: tensor([[-0.0440,  0.0056, -0.0255,  ..., -0.0225, -0.0019, -0.0091]]),\n",
       " 570: tensor([[-0.0215,  0.0120, -0.0055,  ..., -0.0232, -0.0173, -0.0277]]),\n",
       " 571: tensor([[-0.0458,  0.0128, -0.0159,  ..., -0.0451, -0.0157, -0.0093]]),\n",
       " 572: tensor([[-0.0092,  0.0023,  0.0021,  ..., -0.0307, -0.0103, -0.0293]]),\n",
       " 573: tensor([[-0.0404, -0.0186, -0.0143,  ..., -0.0432, -0.0113, -0.0290]]),\n",
       " 574: tensor([[-0.0165,  0.0121, -0.0113,  ..., -0.0254, -0.0196, -0.0067]]),\n",
       " 575: tensor([[-0.0389, -0.0033, -0.0218,  ..., -0.0305, -0.0290, -0.0167]]),\n",
       " 576: tensor([[-0.0358, -0.0015, -0.0230,  ..., -0.0153, -0.0195, -0.0095]]),\n",
       " 577: tensor([[-0.0448, -0.0150, -0.0244,  ..., -0.0252, -0.0222, -0.0059]]),\n",
       " 578: tensor([[-0.0305, -0.0056, -0.0103,  ..., -0.0481, -0.0121, -0.0183]]),\n",
       " 579: tensor([[-0.0482, -0.0069, -0.0288,  ..., -0.0314, -0.0275, -0.0413]]),\n",
       " 580: tensor([[-0.0403, -0.0171, -0.0051,  ..., -0.0323, -0.0120, -0.0206]]),\n",
       " 581: tensor([[-0.0324, -0.0005, -0.0023,  ..., -0.0204, -0.0040, -0.0073]]),\n",
       " 582: tensor([[-0.0304, -0.0083, -0.0181,  ..., -0.0243, -0.0028, -0.0334]]),\n",
       " 583: tensor([[-0.0495,  0.0107, -0.0076,  ..., -0.0189, -0.0126, -0.0065]]),\n",
       " 584: tensor([[-0.0143,  0.0171, -0.0043,  ..., -0.0249, -0.0235, -0.0293]]),\n",
       " 585: tensor([[-0.0530, -0.0211, -0.0349,  ..., -0.0197, -0.0070, -0.0253]]),\n",
       " 586: tensor([[-0.0317,  0.0105, -0.0234,  ..., -0.0244, -0.0070, -0.0434]]),\n",
       " 587: tensor([[-0.0410,  0.0050, -0.0063,  ..., -0.0102, -0.0151, -0.0204]]),\n",
       " 588: tensor([[-0.0461, -0.0078, -0.0069,  ..., -0.0112, -0.0065, -0.0328]]),\n",
       " 589: tensor([[-0.0259,  0.0242,  0.0078,  ..., -0.0435, -0.0121, -0.0240]]),\n",
       " 590: tensor([[-0.0242, -0.0095, -0.0210,  ..., -0.0255, -0.0054, -0.0301]]),\n",
       " 591: tensor([[-0.0347, -0.0107, -0.0208,  ..., -0.0258, -0.0061, -0.0203]]),\n",
       " 592: tensor([[-0.0349,  0.0125, -0.0041,  ..., -0.0194, -0.0156, -0.0255]]),\n",
       " 593: tensor([[-0.0424,  0.0019, -0.0182,  ..., -0.0317, -0.0259, -0.0114]]),\n",
       " 594: tensor([[-0.0366, -0.0044, -0.0223,  ..., -0.0353, -0.0130, -0.0190]]),\n",
       " 595: tensor([[-0.0347, -0.0058,  0.0011,  ..., -0.0205, -0.0099, -0.0217]]),\n",
       " 596: tensor([[-0.0487, -0.0131, -0.0260,  ..., -0.0141, -0.0257, -0.0499]]),\n",
       " 597: tensor([[-0.0320, -0.0097, -0.0152,  ..., -0.0338, -0.0068, -0.0267]]),\n",
       " 598: tensor([[-4.4726e-02, -5.8031e-04,  5.3092e-05,  ..., -7.6848e-03,\n",
       "          -1.2920e-02, -2.6828e-02]]),\n",
       " 599: tensor([[-0.0237, -0.0204, -0.0347,  ..., -0.0294, -0.0020, -0.0602]]),\n",
       " 600: tensor([[-0.0284,  0.0151, -0.0196,  ..., -0.0236, -0.0176, -0.0296]]),\n",
       " 601: tensor([[-0.0393, -0.0044, -0.0291,  ..., -0.0406, -0.0272, -0.0157]]),\n",
       " 602: tensor([[-0.0305, -0.0068, -0.0119,  ..., -0.0179, -0.0174, -0.0167]]),\n",
       " 603: tensor([[-0.0153,  0.0057, -0.0220,  ..., -0.0184, -0.0150, -0.0325]]),\n",
       " 604: tensor([[-0.0478, -0.0114, -0.0143,  ..., -0.0048, -0.0044, -0.0293]]),\n",
       " 605: tensor([[-0.0458, -0.0093, -0.0021,  ..., -0.0189, -0.0165, -0.0116]]),\n",
       " 606: tensor([[-0.0532, -0.0027, -0.0199,  ..., -0.0276, -0.0069, -0.0270]]),\n",
       " 607: tensor([[-0.0349,  0.0245, -0.0050,  ..., -0.0247, -0.0111, -0.0035]]),\n",
       " 608: tensor([[-0.0375, -0.0109, -0.0015,  ..., -0.0234, -0.0056, -0.0180]]),\n",
       " 609: tensor([[-0.0312, -0.0112, -0.0371,  ..., -0.0186, -0.0164, -0.0270]]),\n",
       " 610: tensor([[-0.0330, -0.0100, -0.0376,  ..., -0.0275, -0.0112, -0.0329]]),\n",
       " 611: tensor([[-0.0332,  0.0001, -0.0027,  ..., -0.0275, -0.0203, -0.0154]]),\n",
       " 612: tensor([[-0.0414,  0.0135, -0.0121,  ..., -0.0297, -0.0118, -0.0363]]),\n",
       " 613: tensor([[-0.0258, -0.0014, -0.0201,  ..., -0.0330, -0.0041, -0.0235]]),\n",
       " 614: tensor([[-0.0308,  0.0039, -0.0160,  ..., -0.0313, -0.0095, -0.0291]]),\n",
       " 615: tensor([[-0.0360, -0.0059, -0.0147,  ..., -0.0426, -0.0100, -0.0317]]),\n",
       " 616: tensor([[-0.0533, -0.0096, -0.0104,  ..., -0.0318, -0.0105, -0.0242]]),\n",
       " 617: tensor([[-0.0430, -0.0044, -0.0252,  ..., -0.0293, -0.0113, -0.0145]]),\n",
       " 618: tensor([[-0.0376, -0.0100, -0.0201,  ..., -0.0167, -0.0133, -0.0375]]),\n",
       " 619: tensor([[-0.0320, -0.0086, -0.0112,  ..., -0.0336, -0.0073, -0.0167]]),\n",
       " 620: tensor([[-0.0330,  0.0090,  0.0070,  ..., -0.0152, -0.0047, -0.0196]]),\n",
       " 621: tensor([[-0.0248, -0.0073, -0.0077,  ..., -0.0146, -0.0103, -0.0299]]),\n",
       " 622: tensor([[-0.0294, -0.0063, -0.0068,  ..., -0.0307,  0.0012, -0.0376]]),\n",
       " 623: tensor([[-0.0192,  0.0053, -0.0044,  ..., -0.0473, -0.0107, -0.0153]]),\n",
       " 624: tensor([[-0.0359, -0.0029, -0.0090,  ..., -0.0180, -0.0139, -0.0182]]),\n",
       " 625: tensor([[-0.0339,  0.0119, -0.0392,  ..., -0.0441, -0.0021, -0.0306]]),\n",
       " 626: tensor([[-0.0501,  0.0158, -0.0247,  ..., -0.0092, -0.0172, -0.0273]]),\n",
       " 627: tensor([[-0.0263, -0.0018,  0.0151,  ..., -0.0278, -0.0059, -0.0243]]),\n",
       " 628: tensor([[-0.0354,  0.0061, -0.0243,  ..., -0.0203, -0.0195, -0.0389]]),\n",
       " 629: tensor([[-0.0275,  0.0037, -0.0121,  ..., -0.0371, -0.0188, -0.0351]]),\n",
       " 630: tensor([[-0.0214, -0.0113, -0.0146,  ..., -0.0353, -0.0073, -0.0266]]),\n",
       " 631: tensor([[-0.0371, -0.0013, -0.0199,  ..., -0.0350, -0.0140, -0.0268]]),\n",
       " 632: tensor([[-0.0477,  0.0019, -0.0278,  ..., -0.0443, -0.0038, -0.0234]]),\n",
       " 633: tensor([[-0.0462,  0.0006, -0.0266,  ..., -0.0439, -0.0099, -0.0203]]),\n",
       " 634: tensor([[-0.0246,  0.0063, -0.0181,  ..., -0.0322, -0.0037, -0.0322]]),\n",
       " 635: tensor([[-0.0349, -0.0138,  0.0018,  ..., -0.0251, -0.0198, -0.0103]]),\n",
       " 636: tensor([[-0.0348,  0.0056, -0.0237,  ..., -0.0161, -0.0237, -0.0266]]),\n",
       " 637: tensor([[-0.0363, -0.0037, -0.0105,  ..., -0.0269, -0.0132, -0.0266]]),\n",
       " 638: tensor([[-0.0426, -0.0114, -0.0115,  ..., -0.0224, -0.0233, -0.0349]]),\n",
       " 639: tensor([[-0.0255, -0.0136, -0.0240,  ..., -0.0073, -0.0105, -0.0175]]),\n",
       " 640: tensor([[-0.0382,  0.0169, -0.0151,  ..., -0.0420, -0.0047, -0.0454]]),\n",
       " 641: tensor([[-0.0293,  0.0086, -0.0108,  ..., -0.0250, -0.0164, -0.0369]]),\n",
       " 642: tensor([[-0.0366,  0.0097, -0.0275,  ..., -0.0543, -0.0083, -0.0144]]),\n",
       " 643: tensor([[-0.0262,  0.0113, -0.0219,  ..., -0.0277, -0.0120, -0.0346]]),\n",
       " 644: tensor([[-0.0189, -0.0126, -0.0080,  ..., -0.0209, -0.0064, -0.0156]]),\n",
       " 645: tensor([[-0.0373, -0.0018, -0.0111,  ..., -0.0195, -0.0127, -0.0248]]),\n",
       " 646: tensor([[-0.0371,  0.0050, -0.0100,  ..., -0.0464, -0.0098, -0.0196]]),\n",
       " 647: tensor([[-0.0286,  0.0140, -0.0118,  ..., -0.0202, -0.0198, -0.0213]]),\n",
       " 648: tensor([[-0.0491, -0.0163, -0.0217,  ..., -0.0319, -0.0095, -0.0373]]),\n",
       " 649: tensor([[-0.0533, -0.0106, -0.0103,  ..., -0.0159, -0.0272, -0.0275]]),\n",
       " 650: tensor([[-0.0412,  0.0099, -0.0061,  ..., -0.0322, -0.0223, -0.0132]]),\n",
       " 651: tensor([[-0.0203, -0.0093, -0.0071,  ..., -0.0192, -0.0238, -0.0267]]),\n",
       " 652: tensor([[-0.0242, -0.0015, -0.0162,  ..., -0.0297, -0.0084, -0.0194]]),\n",
       " 653: tensor([[-0.0296,  0.0012, -0.0028,  ..., -0.0240, -0.0029, -0.0107]]),\n",
       " 654: tensor([[-0.0405, -0.0109, -0.0210,  ..., -0.0271, -0.0063, -0.0240]]),\n",
       " 655: tensor([[-0.0366, -0.0083, -0.0007,  ..., -0.0304, -0.0090, -0.0248]]),\n",
       " 656: tensor([[-0.0208,  0.0049, -0.0162,  ..., -0.0334,  0.0047, -0.0202]]),\n",
       " 657: tensor([[-0.0212, -0.0037, -0.0266,  ..., -0.0432, -0.0002, -0.0122]]),\n",
       " 658: tensor([[-0.0442,  0.0008, -0.0254,  ..., -0.0183, -0.0022, -0.0162]]),\n",
       " 659: tensor([[-0.0251, -0.0155, -0.0196,  ..., -0.0088, -0.0049, -0.0174]]),\n",
       " 660: tensor([[-0.0217, -0.0023, -0.0173,  ..., -0.0267,  0.0061, -0.0375]]),\n",
       " 661: tensor([[-0.0443, -0.0048, -0.0042,  ..., -0.0382, -0.0117, -0.0156]]),\n",
       " 662: tensor([[-0.0368,  0.0036, -0.0083,  ..., -0.0261, -0.0147, -0.0090]]),\n",
       " 663: tensor([[-0.0454, -0.0063, -0.0133,  ..., -0.0211, -0.0175, -0.0313]]),\n",
       " 664: tensor([[-0.0278,  0.0056, -0.0076,  ..., -0.0235,  0.0133, -0.0313]]),\n",
       " 665: tensor([[-0.0322,  0.0018, -0.0344,  ..., -0.0331, -0.0046, -0.0349]]),\n",
       " 666: tensor([[-0.0343, -0.0004, -0.0190,  ..., -0.0145,  0.0077, -0.0325]]),\n",
       " 667: tensor([[-0.0391, -0.0092, -0.0296,  ..., -0.0187, -0.0003, -0.0367]]),\n",
       " 668: tensor([[-0.0515, -0.0053, -0.0206,  ..., -0.0338,  0.0039, -0.0385]]),\n",
       " 669: tensor([[-0.0480, -0.0067, -0.0161,  ..., -0.0332,  0.0038, -0.0248]]),\n",
       " 670: tensor([[-0.0338,  0.0046, -0.0189,  ..., -0.0347, -0.0147, -0.0275]]),\n",
       " 671: tensor([[-0.0530, -0.0044, -0.0425,  ..., -0.0099,  0.0037, -0.0168]]),\n",
       " 672: tensor([[-0.0354,  0.0039, -0.0111,  ..., -0.0285, -0.0145, -0.0371]]),\n",
       " 673: tensor([[-0.0307,  0.0173,  0.0045,  ..., -0.0145, -0.0102, -0.0384]]),\n",
       " 674: tensor([[-0.0438,  0.0071, -0.0053,  ..., -0.0483, -0.0112, -0.0068]]),\n",
       " 675: tensor([[-0.0330,  0.0085, -0.0088,  ..., -0.0330,  0.0029, -0.0130]]),\n",
       " 676: tensor([[-0.0344, -0.0149, -0.0234,  ..., -0.0264, -0.0242, -0.0233]]),\n",
       " 677: tensor([[-0.0108,  0.0049, -0.0122,  ..., -0.0246, -0.0055, -0.0087]]),\n",
       " 678: tensor([[-3.7245e-02,  2.4523e-05, -8.3748e-03,  ..., -3.3045e-02,\n",
       "          -1.4569e-02, -3.0449e-02]]),\n",
       " 679: tensor([[-0.0377, -0.0010, -0.0275,  ..., -0.0414, -0.0177, -0.0344]]),\n",
       " 680: tensor([[-0.0277,  0.0110, -0.0104,  ..., -0.0192, -0.0022, -0.0324]]),\n",
       " 681: tensor([[-0.0248, -0.0098, -0.0152,  ..., -0.0231, -0.0029, -0.0398]]),\n",
       " 682: tensor([[-0.0304,  0.0103, -0.0055,  ..., -0.0221, -0.0115, -0.0498]]),\n",
       " 683: tensor([[-0.0376,  0.0082, -0.0324,  ..., -0.0429, -0.0093, -0.0285]]),\n",
       " 684: tensor([[-0.0356,  0.0028, -0.0080,  ..., -0.0055, -0.0037, -0.0094]]),\n",
       " 685: tensor([[-0.0221,  0.0132, -0.0020,  ..., -0.0187, -0.0265, -0.0175]]),\n",
       " 686: tensor([[-0.0344, -0.0039, -0.0032,  ..., -0.0369, -0.0150, -0.0294]]),\n",
       " 687: tensor([[-0.0364, -0.0105, -0.0137,  ..., -0.0299, -0.0171, -0.0101]]),\n",
       " 688: tensor([[-0.0240,  0.0098, -0.0094,  ..., -0.0301, -0.0054, -0.0573]]),\n",
       " 689: tensor([[-0.0310, -0.0027, -0.0269,  ..., -0.0333, -0.0137, -0.0094]]),\n",
       " 690: tensor([[-0.0366, -0.0149, -0.0236,  ..., -0.0293, -0.0023, -0.0418]]),\n",
       " 691: tensor([[-0.0348,  0.0166, -0.0073,  ..., -0.0475, -0.0328, -0.0133]]),\n",
       " 692: tensor([[-0.0361, -0.0140, -0.0207,  ..., -0.0399, -0.0118, -0.0327]]),\n",
       " 693: tensor([[-0.0388, -0.0036, -0.0069,  ..., -0.0162, -0.0052, -0.0344]]),\n",
       " 694: tensor([[-0.0540, -0.0117, -0.0138,  ..., -0.0162, -0.0176, -0.0227]]),\n",
       " 695: tensor([[-0.0402,  0.0031, -0.0150,  ..., -0.0218,  0.0060, -0.0377]]),\n",
       " 696: tensor([[-0.0433,  0.0003, -0.0140,  ..., -0.0346, -0.0236, -0.0275]]),\n",
       " 697: tensor([[-0.0357,  0.0041, -0.0016,  ..., -0.0214, -0.0192, -0.0393]]),\n",
       " 698: tensor([[-0.0227, -0.0051, -0.0036,  ..., -0.0037, -0.0118, -0.0274]]),\n",
       " 699: tensor([[-0.0346, -0.0015, -0.0046,  ..., -0.0305, -0.0115, -0.0209]]),\n",
       " 700: tensor([[-0.0249, -0.0059, -0.0178,  ..., -0.0064,  0.0026, -0.0221]]),\n",
       " 701: tensor([[-0.0404,  0.0004, -0.0015,  ..., -0.0261, -0.0109, -0.0263]]),\n",
       " 702: tensor([[-0.0177,  0.0111,  0.0015,  ..., -0.0293, -0.0095, -0.0312]]),\n",
       " 703: tensor([[-0.0216,  0.0025, -0.0008,  ..., -0.0252,  0.0012, -0.0318]]),\n",
       " 704: tensor([[-0.0460,  0.0046, -0.0099,  ..., -0.0336, -0.0112, -0.0318]]),\n",
       " 705: tensor([[-0.0559, -0.0047, -0.0274,  ..., -0.0256, -0.0154, -0.0428]]),\n",
       " 706: tensor([[-0.0295, -0.0080, -0.0166,  ..., -0.0343, -0.0144, -0.0196]]),\n",
       " 707: tensor([[-0.0289,  0.0147, -0.0169,  ..., -0.0376, -0.0304, -0.0192]]),\n",
       " 708: tensor([[-0.0485, -0.0081, -0.0318,  ..., -0.0314, -0.0225, -0.0402]]),\n",
       " 709: tensor([[-0.0361, -0.0180, -0.0212,  ..., -0.0178, -0.0317, -0.0219]]),\n",
       " 710: tensor([[-0.0541, -0.0043, -0.0128,  ..., -0.0292, -0.0310, -0.0170]]),\n",
       " 711: tensor([[-0.0280, -0.0152,  0.0056,  ..., -0.0170, -0.0175, -0.0361]]),\n",
       " 712: tensor([[-0.0086,  0.0108, -0.0051,  ..., -0.0250, -0.0251, -0.0356]]),\n",
       " 713: tensor([[-0.0366,  0.0071, -0.0116,  ..., -0.0341, -0.0037, -0.0214]]),\n",
       " 714: tensor([[-0.0424, -0.0077, -0.0356,  ..., -0.0389, -0.0147, -0.0213]]),\n",
       " 715: tensor([[-0.0499,  0.0029, -0.0389,  ..., -0.0295,  0.0014, -0.0321]]),\n",
       " 716: tensor([[-0.0359, -0.0215, -0.0386,  ..., -0.0403,  0.0152, -0.0525]]),\n",
       " 717: tensor([[-0.0295, -0.0094, -0.0127,  ..., -0.0222,  0.0048, -0.0410]]),\n",
       " 718: tensor([[-0.0344, -0.0088, -0.0113,  ..., -0.0223, -0.0189, -0.0055]]),\n",
       " 719: tensor([[-0.0415, -0.0051, -0.0185,  ..., -0.0294, -0.0062, -0.0109]]),\n",
       " 720: tensor([[-0.0468, -0.0057, -0.0226,  ..., -0.0313, -0.0185, -0.0280]]),\n",
       " 721: tensor([[-0.0232,  0.0012,  0.0015,  ..., -0.0205, -0.0153, -0.0133]]),\n",
       " 722: tensor([[-0.0338, -0.0226, -0.0369,  ..., -0.0406, -0.0063, -0.0409]]),\n",
       " 723: tensor([[-0.0260, -0.0028, -0.0128,  ..., -0.0403, -0.0068, -0.0268]]),\n",
       " 724: tensor([[-0.0431,  0.0013, -0.0171,  ..., -0.0330, -0.0054, -0.0090]]),\n",
       " 725: tensor([[-0.0384, -0.0057, -0.0071,  ..., -0.0156, -0.0037, -0.0374]]),\n",
       " 726: tensor([[-0.0382,  0.0080, -0.0216,  ..., -0.0206, -0.0200, -0.0193]]),\n",
       " 727: tensor([[-0.0253,  0.0053, -0.0034,  ..., -0.0251, -0.0105, -0.0092]]),\n",
       " 728: tensor([[-0.0494, -0.0095, -0.0165,  ..., -0.0256, -0.0120, -0.0173]]),\n",
       " 729: tensor([[-0.0496, -0.0037, -0.0115,  ..., -0.0243, -0.0114, -0.0233]]),\n",
       " 730: tensor([[-0.0276, -0.0031, -0.0164,  ..., -0.0416, -0.0083, -0.0234]]),\n",
       " 731: tensor([[-0.0234,  0.0157, -0.0091,  ..., -0.0345, -0.0019, -0.0241]]),\n",
       " 732: tensor([[-0.0432, -0.0016, -0.0052,  ..., -0.0137, -0.0198, -0.0329]]),\n",
       " 733: tensor([[-0.0391,  0.0049, -0.0080,  ..., -0.0159, -0.0013, -0.0143]]),\n",
       " 734: tensor([[-0.0288,  0.0136, -0.0155,  ..., -0.0149, -0.0140, -0.0353]]),\n",
       " 735: tensor([[-0.0421,  0.0048, -0.0273,  ..., -0.0267,  0.0162, -0.0187]]),\n",
       " 736: tensor([[-0.0448, -0.0013, -0.0174,  ..., -0.0238, -0.0147, -0.0353]]),\n",
       " 737: tensor([[-0.0479, -0.0134, -0.0273,  ..., -0.0245,  0.0045, -0.0279]]),\n",
       " 738: tensor([[-0.0434,  0.0029, -0.0096,  ..., -0.0432, -0.0124, -0.0141]]),\n",
       " 739: tensor([[-0.0283, -0.0027,  0.0041,  ..., -0.0188, -0.0087, -0.0416]]),\n",
       " 740: tensor([[-0.0240, -0.0100, -0.0137,  ..., -0.0261, -0.0164, -0.0234]]),\n",
       " 741: tensor([[-0.0334, -0.0103, -0.0289,  ..., -0.0298, -0.0116, -0.0202]]),\n",
       " 742: tensor([[-0.0522,  0.0034, -0.0422,  ..., -0.0117, -0.0105, -0.0441]]),\n",
       " 743: tensor([[-0.0121,  0.0060,  0.0013,  ..., -0.0287, -0.0065, -0.0315]]),\n",
       " 744: tensor([[-0.0463,  0.0105, -0.0137,  ..., -0.0240, -0.0221, -0.0145]]),\n",
       " 745: tensor([[-0.0220, -0.0159, -0.0239,  ..., -0.0290, -0.0077, -0.0397]]),\n",
       " 746: tensor([[-0.0401,  0.0077, -0.0354,  ..., -0.0169, -0.0209, -0.0288]]),\n",
       " 747: tensor([[-0.0233,  0.0195, -0.0093,  ..., -0.0243, -0.0147, -0.0238]]),\n",
       " 748: tensor([[-0.0234, -0.0075, -0.0216,  ..., -0.0319, -0.0174, -0.0117]]),\n",
       " 749: tensor([[-0.0539, -0.0109, -0.0187,  ..., -0.0531, -0.0190, -0.0174]]),\n",
       " 750: tensor([[-0.0296,  0.0075, -0.0157,  ..., -0.0323, -0.0028, -0.0374]]),\n",
       " 751: tensor([[-0.0422,  0.0063, -0.0176,  ..., -0.0344, -0.0164, -0.0206]]),\n",
       " 752: tensor([[-0.0309,  0.0049, -0.0060,  ..., -0.0151, -0.0115, -0.0114]]),\n",
       " 753: tensor([[-0.0393,  0.0036, -0.0099,  ..., -0.0292, -0.0105, -0.0226]]),\n",
       " 754: tensor([[-0.0458,  0.0032, -0.0209,  ..., -0.0606, -0.0183, -0.0227]]),\n",
       " 755: tensor([[-0.0368, -0.0062, -0.0250,  ..., -0.0315, -0.0080, -0.0344]]),\n",
       " 756: tensor([[-0.0265, -0.0030, -0.0129,  ..., -0.0363, -0.0100, -0.0078]]),\n",
       " 757: tensor([[-0.0238, -0.0058, -0.0362,  ..., -0.0347, -0.0176, -0.0310]]),\n",
       " 758: tensor([[-0.0423, -0.0059, -0.0151,  ..., -0.0479, -0.0247, -0.0274]]),\n",
       " 759: tensor([[-0.0400, -0.0238, -0.0321,  ..., -0.0449, -0.0033, -0.0345]]),\n",
       " 760: tensor([[-3.3441e-02,  5.0609e-05, -1.9195e-02,  ..., -1.0741e-02,\n",
       "          -1.7409e-03, -2.2906e-02]]),\n",
       " 761: tensor([[-0.0226,  0.0015, -0.0211,  ..., -0.0241, -0.0008, -0.0317]]),\n",
       " 762: tensor([[-0.0473, -0.0014, -0.0203,  ..., -0.0157, -0.0102, -0.0354]]),\n",
       " 763: tensor([[-0.0471, -0.0076, -0.0206,  ..., -0.0300, -0.0177, -0.0126]]),\n",
       " 764: tensor([[-0.0343,  0.0125, -0.0105,  ..., -0.0461, -0.0303, -0.0085]]),\n",
       " 765: tensor([[-0.0278, -0.0028, -0.0018,  ..., -0.0319, -0.0059, -0.0236]]),\n",
       " 766: tensor([[-0.0260,  0.0017, -0.0084,  ..., -0.0356, -0.0122, -0.0164]]),\n",
       " 767: tensor([[-0.0479, -0.0009, -0.0143,  ..., -0.0198, -0.0063, -0.0139]]),\n",
       " 768: tensor([[-0.0344,  0.0019, -0.0158,  ..., -0.0325,  0.0043, -0.0063]]),\n",
       " 769: tensor([[-0.0431,  0.0051, -0.0172,  ..., -0.0334, -0.0252, -0.0268]]),\n",
       " 770: tensor([[-0.0444, -0.0032, -0.0199,  ..., -0.0247, -0.0101, -0.0198]]),\n",
       " 771: tensor([[-0.0373,  0.0021, -0.0046,  ..., -0.0237, -0.0070, -0.0257]]),\n",
       " 772: tensor([[-0.0279, -0.0070, -0.0072,  ..., -0.0349, -0.0142, -0.0356]]),\n",
       " 773: tensor([[-0.0283,  0.0030, -0.0267,  ..., -0.0518, -0.0207, -0.0186]]),\n",
       " 774: tensor([[-0.0492, -0.0009,  0.0032,  ..., -0.0123, -0.0281, -0.0344]]),\n",
       " 775: tensor([[-0.0336, -0.0087, -0.0195,  ..., -0.0215, -0.0044, -0.0301]]),\n",
       " 776: tensor([[-0.0214,  0.0121, -0.0034,  ..., -0.0302, -0.0042, -0.0339]]),\n",
       " 777: tensor([[-0.0473, -0.0122, -0.0051,  ..., -0.0174, -0.0016, -0.0256]]),\n",
       " 778: tensor([[-0.0488, -0.0013,  0.0034,  ..., -0.0174, -0.0072, -0.0319]]),\n",
       " 779: tensor([[-0.0504, -0.0048, -0.0110,  ..., -0.0111, -0.0180, -0.0234]]),\n",
       " 780: tensor([[-0.0409,  0.0125, -0.0016,  ..., -0.0268, -0.0017, -0.0285]]),\n",
       " 781: tensor([[-0.0348, -0.0034, -0.0141,  ..., -0.0293, -0.0194, -0.0218]]),\n",
       " 782: tensor([[-0.0367, -0.0102, -0.0215,  ..., -0.0338, -0.0050, -0.0351]]),\n",
       " 783: tensor([[-0.0263,  0.0132, -0.0101,  ..., -0.0222, -0.0098, -0.0335]]),\n",
       " 784: tensor([[-0.0380,  0.0006, -0.0163,  ..., -0.0312, -0.0194, -0.0183]]),\n",
       " 785: tensor([[-0.0447, -0.0056, -0.0076,  ..., -0.0369, -0.0041, -0.0206]]),\n",
       " 786: tensor([[-0.0403, -0.0025,  0.0018,  ..., -0.0322, -0.0258, -0.0214]]),\n",
       " 787: tensor([[-0.0508, -0.0176, -0.0342,  ..., -0.0307,  0.0018, -0.0326]]),\n",
       " 788: tensor([[-0.0272,  0.0114, -0.0144,  ..., -0.0315, -0.0253, -0.0269]]),\n",
       " 789: tensor([[-0.0273,  0.0010, -0.0128,  ..., -0.0131, -0.0071, -0.0220]]),\n",
       " 790: tensor([[-0.0380, -0.0059, -0.0191,  ..., -0.0342, -0.0190, -0.0352]]),\n",
       " 791: tensor([[-0.0424,  0.0050, -0.0030,  ..., -0.0281, -0.0153, -0.0292]]),\n",
       " 792: tensor([[-0.0402,  0.0083,  0.0111,  ..., -0.0369, -0.0077, -0.0252]]),\n",
       " 793: tensor([[-0.0336,  0.0027, -0.0341,  ..., -0.0491,  0.0088, -0.0326]]),\n",
       " 794: tensor([[-0.0497, -0.0046, -0.0183,  ..., -0.0162, -0.0187, -0.0305]]),\n",
       " 795: tensor([[-0.0480, -0.0022, -0.0158,  ..., -0.0300, -0.0164, -0.0246]]),\n",
       " 796: tensor([[-0.0208,  0.0105, -0.0135,  ..., -0.0323, -0.0070, -0.0109]]),\n",
       " 797: tensor([[-0.0458,  0.0144, -0.0023,  ..., -0.0319, -0.0131, -0.0309]]),\n",
       " 798: tensor([[-0.0177, -0.0053, -0.0241,  ..., -0.0168, -0.0051, -0.0382]]),\n",
       " 799: tensor([[-0.0447,  0.0032, -0.0214,  ..., -0.0424, -0.0125, -0.0211]]),\n",
       " 800: tensor([[-0.0381,  0.0036, -0.0228,  ..., -0.0338,  0.0059, -0.0195]]),\n",
       " 801: tensor([[-0.0354,  0.0038, -0.0174,  ..., -0.0169, -0.0040, -0.0271]]),\n",
       " 802: tensor([[-0.0367,  0.0007, -0.0206,  ..., -0.0359, -0.0163, -0.0252]]),\n",
       " 803: tensor([[-0.0268,  0.0200, -0.0136,  ..., -0.0288, -0.0104, -0.0219]]),\n",
       " 804: tensor([[-0.0362,  0.0084, -0.0310,  ..., -0.0286, -0.0138, -0.0383]]),\n",
       " 805: tensor([[-0.0408, -0.0047, -0.0056,  ..., -0.0368, -0.0187, -0.0411]]),\n",
       " 806: tensor([[-0.0461,  0.0080, -0.0075,  ..., -0.0337, -0.0029, -0.0131]]),\n",
       " 807: tensor([[-0.0315, -0.0130, -0.0128,  ..., -0.0522, -0.0076, -0.0169]]),\n",
       " 808: tensor([[-0.0384,  0.0098, -0.0100,  ..., -0.0339, -0.0176, -0.0082]]),\n",
       " 809: tensor([[-0.0464, -0.0170, -0.0486,  ..., -0.0295, -0.0363, -0.0241]]),\n",
       " 810: tensor([[-0.0308, -0.0277, -0.0219,  ..., -0.0124, -0.0017, -0.0329]]),\n",
       " 811: tensor([[-0.0360, -0.0148, -0.0206,  ..., -0.0506, -0.0014, -0.0353]]),\n",
       " 812: tensor([[-0.0353, -0.0074, -0.0078,  ..., -0.0095, -0.0052, -0.0242]]),\n",
       " 813: tensor([[-0.0357,  0.0003, -0.0232,  ..., -0.0399, -0.0162, -0.0146]]),\n",
       " 814: tensor([[-0.0480,  0.0007, -0.0210,  ..., -0.0283, -0.0305, -0.0134]]),\n",
       " 815: tensor([[-0.0367,  0.0067, -0.0046,  ..., -0.0241,  0.0073, -0.0254]]),\n",
       " 816: tensor([[-3.3286e-02,  4.2725e-05, -3.1488e-03,  ..., -1.9399e-02,\n",
       "          -5.3007e-04, -2.2674e-02]]),\n",
       " 817: tensor([[-0.0399, -0.0086,  0.0147,  ..., -0.0382, -0.0061, -0.0184]]),\n",
       " 818: tensor([[-0.0302, -0.0002, -0.0296,  ..., -0.0448, -0.0275, -0.0318]]),\n",
       " 819: tensor([[-0.0497, -0.0015, -0.0141,  ..., -0.0254, -0.0152, -0.0282]]),\n",
       " 820: tensor([[-0.0343,  0.0027, -0.0145,  ..., -0.0346,  0.0069, -0.0212]]),\n",
       " 821: tensor([[-0.0476,  0.0055, -0.0074,  ..., -0.0130, -0.0194, -0.0412]]),\n",
       " 822: tensor([[-0.0472,  0.0041, -0.0263,  ..., -0.0330, -0.0161, -0.0192]]),\n",
       " 823: tensor([[-0.0324, -0.0094, -0.0172,  ..., -0.0372, -0.0078, -0.0260]]),\n",
       " 824: tensor([[-0.0269, -0.0115, -0.0287,  ..., -0.0279, -0.0215, -0.0241]]),\n",
       " 825: tensor([[-0.0298,  0.0027, -0.0339,  ..., -0.0296, -0.0098, -0.0244]]),\n",
       " 826: tensor([[-0.0486, -0.0021, -0.0206,  ..., -0.0214, -0.0246, -0.0265]]),\n",
       " 827: tensor([[-0.0340, -0.0047, -0.0171,  ..., -0.0323, -0.0053, -0.0297]]),\n",
       " 828: tensor([[-0.0275, -0.0100,  0.0006,  ..., -0.0296, -0.0016, -0.0524]]),\n",
       " 829: tensor([[-4.7314e-02, -5.2942e-03, -6.7976e-05,  ..., -3.1386e-02,\n",
       "          -3.0946e-02, -1.7884e-02]]),\n",
       " 830: tensor([[-0.0286, -0.0116, -0.0255,  ..., -0.0369, -0.0192, -0.0196]]),\n",
       " 831: tensor([[-0.0302,  0.0050, -0.0190,  ..., -0.0305, -0.0019, -0.0125]]),\n",
       " 832: tensor([[-0.0267,  0.0175, -0.0064,  ..., -0.0490, -0.0238, -0.0162]]),\n",
       " 833: tensor([[-0.0254,  0.0114, -0.0195,  ..., -0.0123, -0.0065, -0.0271]]),\n",
       " 834: tensor([[-0.0432, -0.0057, -0.0150,  ..., -0.0134, -0.0248, -0.0311]]),\n",
       " 835: tensor([[-0.0408,  0.0195,  0.0137,  ..., -0.0364, -0.0042, -0.0146]]),\n",
       " 836: tensor([[-0.0353, -0.0007, -0.0070,  ..., -0.0096, -0.0236, -0.0159]]),\n",
       " 837: tensor([[-0.0422,  0.0121, -0.0168,  ..., -0.0457, -0.0194, -0.0092]]),\n",
       " 838: tensor([[-0.0426, -0.0056, -0.0263,  ..., -0.0332, -0.0151, -0.0424]]),\n",
       " 839: tensor([[-0.0468, -0.0015, -0.0071,  ..., -0.0308, -0.0288, -0.0311]]),\n",
       " 840: tensor([[-0.0331,  0.0040, -0.0295,  ..., -0.0392, -0.0184, -0.0313]]),\n",
       " 841: tensor([[-0.0310,  0.0072, -0.0216,  ..., -0.0299,  0.0027, -0.0276]]),\n",
       " 842: tensor([[-0.0284, -0.0089, -0.0265,  ..., -0.0284, -0.0124, -0.0260]]),\n",
       " 843: tensor([[-0.0513, -0.0139, -0.0190,  ..., -0.0322, -0.0093, -0.0311]]),\n",
       " 844: tensor([[-0.0320,  0.0124, -0.0019,  ..., -0.0262, -0.0136, -0.0187]]),\n",
       " 845: tensor([[-0.0241, -0.0031, -0.0141,  ..., -0.0316, -0.0197, -0.0149]]),\n",
       " 846: tensor([[-0.0526,  0.0019, -0.0222,  ..., -0.0431, -0.0081, -0.0263]]),\n",
       " 847: tensor([[-0.0197,  0.0113,  0.0045,  ..., -0.0083, -0.0244, -0.0202]]),\n",
       " 848: tensor([[-0.0333, -0.0060, -0.0163,  ..., -0.0235, -0.0032, -0.0216]]),\n",
       " 849: tensor([[-0.0344,  0.0227, -0.0065,  ..., -0.0056, -0.0127, -0.0291]]),\n",
       " 850: tensor([[-0.0399, -0.0019, -0.0176,  ..., -0.0240, -0.0228, -0.0174]]),\n",
       " 851: tensor([[-0.0359,  0.0051,  0.0042,  ..., -0.0128, -0.0124, -0.0175]]),\n",
       " 852: tensor([[-0.0233, -0.0037, -0.0198,  ..., -0.0226, -0.0069, -0.0211]]),\n",
       " 853: tensor([[-0.0335,  0.0010, -0.0112,  ..., -0.0179, -0.0074, -0.0279]]),\n",
       " 854: tensor([[-0.0205, -0.0144, -0.0238,  ..., -0.0261, -0.0091, -0.0438]]),\n",
       " 855: tensor([[-0.0186, -0.0019, -0.0193,  ..., -0.0405, -0.0207, -0.0274]]),\n",
       " 856: tensor([[-0.0216, -0.0053, -0.0343,  ..., -0.0238, -0.0138, -0.0258]]),\n",
       " 857: tensor([[-0.0484, -0.0036, -0.0134,  ..., -0.0343, -0.0203,  0.0011]]),\n",
       " 858: tensor([[-0.0302, -0.0019, -0.0350,  ..., -0.0251, -0.0100, -0.0256]]),\n",
       " 859: tensor([[-0.0382,  0.0095, -0.0017,  ..., -0.0165, -0.0179, -0.0227]]),\n",
       " 860: tensor([[-0.0302,  0.0211,  0.0062,  ..., -0.0447, -0.0133, -0.0235]]),\n",
       " 861: tensor([[-0.0444,  0.0129, -0.0127,  ..., -0.0362, -0.0229, -0.0279]]),\n",
       " 862: tensor([[-0.0465, -0.0048, -0.0127,  ..., -0.0369, -0.0141, -0.0161]]),\n",
       " 863: tensor([[-0.0497,  0.0109, -0.0205,  ..., -0.0351, -0.0167, -0.0278]]),\n",
       " 864: tensor([[-0.0388,  0.0102, -0.0058,  ..., -0.0198, -0.0227, -0.0242]]),\n",
       " 865: tensor([[-0.0210,  0.0153, -0.0120,  ..., -0.0169, -0.0109, -0.0124]]),\n",
       " 866: tensor([[-0.0452, -0.0123, -0.0267,  ..., -0.0247, -0.0104, -0.0190]]),\n",
       " 867: tensor([[-0.0382,  0.0028, -0.0296,  ..., -0.0259,  0.0037, -0.0274]]),\n",
       " 868: tensor([[-0.0445, -0.0032, -0.0183,  ..., -0.0369, -0.0093, -0.0273]]),\n",
       " 869: tensor([[-0.0437, -0.0190, -0.0246,  ..., -0.0464, -0.0149, -0.0294]]),\n",
       " 870: tensor([[-0.0520, -0.0197, -0.0093,  ..., -0.0063, -0.0096, -0.0332]]),\n",
       " 871: tensor([[-0.0416, -0.0041, -0.0086,  ..., -0.0158, -0.0016, -0.0274]]),\n",
       " 872: tensor([[-0.0452, -0.0139, -0.0249,  ..., -0.0118, -0.0154, -0.0152]]),\n",
       " 873: tensor([[-0.0406, -0.0002, -0.0359,  ..., -0.0453, -0.0120, -0.0278]]),\n",
       " 874: tensor([[-0.0177, -0.0077, -0.0130,  ..., -0.0306, -0.0091, -0.0207]]),\n",
       " 875: tensor([[-0.0400,  0.0207, -0.0203,  ..., -0.0247, -0.0106, -0.0099]]),\n",
       " 876: tensor([[-0.0278,  0.0183, -0.0057,  ..., -0.0168, -0.0342, -0.0089]]),\n",
       " 877: tensor([[-0.0299,  0.0128, -0.0034,  ..., -0.0180, -0.0296, -0.0148]]),\n",
       " 878: tensor([[-0.0556,  0.0011, -0.0180,  ..., -0.0249, -0.0102, -0.0540]]),\n",
       " 879: tensor([[-0.0283,  0.0037, -0.0118,  ..., -0.0304,  0.0004, -0.0288]]),\n",
       " 880: tensor([[-0.0407,  0.0076, -0.0051,  ..., -0.0260, -0.0092, -0.0358]]),\n",
       " 881: tensor([[-0.0568, -0.0108, -0.0279,  ..., -0.0073, -0.0239, -0.0419]]),\n",
       " 882: tensor([[-0.0618, -0.0026, -0.0170,  ..., -0.0262, -0.0108, -0.0357]]),\n",
       " 883: tensor([[-0.0499, -0.0087, -0.0107,  ..., -0.0403, -0.0194, -0.0256]]),\n",
       " 884: tensor([[-0.0281, -0.0006, -0.0013,  ..., -0.0114, -0.0239, -0.0272]]),\n",
       " 885: tensor([[-0.0314, -0.0038, -0.0178,  ..., -0.0213, -0.0079, -0.0333]]),\n",
       " 886: tensor([[-0.0306,  0.0041, -0.0212,  ..., -0.0290, -0.0172, -0.0299]]),\n",
       " 887: tensor([[-0.0312,  0.0056, -0.0095,  ..., -0.0254, -0.0189, -0.0430]]),\n",
       " 888: tensor([[-0.0385,  0.0107, -0.0068,  ..., -0.0426, -0.0253, -0.0214]]),\n",
       " 889: tensor([[-0.0258, -0.0082, -0.0105,  ..., -0.0246, -0.0045, -0.0342]]),\n",
       " 890: tensor([[-0.0162, -0.0013, -0.0142,  ..., -0.0241, -0.0161, -0.0322]]),\n",
       " 891: tensor([[-0.0415, -0.0114, -0.0070,  ..., -0.0271,  0.0040, -0.0092]]),\n",
       " 892: tensor([[-0.0393, -0.0006, -0.0210,  ..., -0.0276, -0.0213, -0.0265]]),\n",
       " 893: tensor([[-0.0413,  0.0009, -0.0298,  ..., -0.0193, -0.0149, -0.0109]]),\n",
       " 894: tensor([[-0.0327, -0.0129, -0.0151,  ..., -0.0269,  0.0165, -0.0377]]),\n",
       " 895: tensor([[-2.6312e-02,  5.2134e-05, -1.5181e-02,  ..., -2.4246e-02,\n",
       "          -1.5035e-03, -1.5548e-02]]),\n",
       " 896: tensor([[-0.0415,  0.0015, -0.0194,  ..., -0.0419, -0.0052, -0.0371]]),\n",
       " 897: tensor([[-0.0350,  0.0073, -0.0166,  ..., -0.0176, -0.0149, -0.0318]]),\n",
       " 898: tensor([[-0.0295,  0.0136, -0.0148,  ..., -0.0334, -0.0123, -0.0366]]),\n",
       " 899: tensor([[-0.0549, -0.0016, -0.0217,  ..., -0.0254, -0.0210, -0.0285]]),\n",
       " 900: tensor([[-0.0330,  0.0039, -0.0194,  ..., -0.0250,  0.0054, -0.0284]]),\n",
       " 901: tensor([[-0.0311, -0.0040, -0.0061,  ..., -0.0259, -0.0150, -0.0230]]),\n",
       " 902: tensor([[-0.0571, -0.0078, -0.0197,  ..., -0.0319, -0.0169, -0.0095]]),\n",
       " 903: tensor([[-0.0414, -0.0137, -0.0226,  ..., -0.0336, -0.0011, -0.0319]]),\n",
       " 904: tensor([[-0.0489, -0.0129, -0.0053,  ..., -0.0164, -0.0215, -0.0261]]),\n",
       " 905: tensor([[-0.0282, -0.0013, -0.0115,  ..., -0.0295, -0.0053, -0.0207]]),\n",
       " 906: tensor([[-0.0346,  0.0035, -0.0014,  ..., -0.0369, -0.0111, -0.0310]]),\n",
       " 907: tensor([[-0.0310, -0.0007, -0.0147,  ..., -0.0326, -0.0058, -0.0244]]),\n",
       " 908: tensor([[-0.0301,  0.0026, -0.0238,  ..., -0.0292, -0.0064, -0.0118]]),\n",
       " 909: tensor([[-0.0244,  0.0154, -0.0161,  ..., -0.0388, -0.0054, -0.0180]]),\n",
       " 910: tensor([[-0.0282,  0.0108, -0.0267,  ..., -0.0408, -0.0140, -0.0161]]),\n",
       " 911: tensor([[-0.0383, -0.0048, -0.0206,  ..., -0.0246,  0.0028, -0.0309]]),\n",
       " 912: tensor([[-0.0475, -0.0155, -0.0174,  ..., -0.0203, -0.0096, -0.0250]]),\n",
       " 913: tensor([[-0.0380,  0.0009, -0.0332,  ..., -0.0214, -0.0063, -0.0319]]),\n",
       " 914: tensor([[-0.0307,  0.0200, -0.0205,  ..., -0.0064, -0.0189, -0.0288]]),\n",
       " 915: tensor([[-0.0278, -0.0054, -0.0072,  ..., -0.0295, -0.0208, -0.0120]]),\n",
       " 916: tensor([[-0.0303, -0.0123, -0.0169,  ..., -0.0366, -0.0038, -0.0145]]),\n",
       " 917: tensor([[-0.0253, -0.0011, -0.0157,  ..., -0.0315, -0.0048, -0.0260]]),\n",
       " 918: tensor([[-0.0394,  0.0126, -0.0188,  ..., -0.0447, -0.0060, -0.0121]]),\n",
       " 919: tensor([[-0.0315,  0.0069, -0.0164,  ..., -0.0330, -0.0119, -0.0153]]),\n",
       " 920: tensor([[-0.0398, -0.0191, -0.0175,  ..., -0.0085, -0.0188, -0.0152]]),\n",
       " 921: tensor([[-0.0337,  0.0047, -0.0064,  ..., -0.0231, -0.0139, -0.0391]]),\n",
       " 922: tensor([[-0.0276, -0.0041, -0.0164,  ..., -0.0259,  0.0068, -0.0324]]),\n",
       " 923: tensor([[-0.0213, -0.0070, -0.0203,  ..., -0.0390, -0.0018, -0.0321]]),\n",
       " 924: tensor([[-0.0287,  0.0009, -0.0122,  ..., -0.0149, -0.0078, -0.0390]]),\n",
       " 925: tensor([[-0.0346,  0.0030, -0.0123,  ..., -0.0122, -0.0106, -0.0325]]),\n",
       " 926: tensor([[-0.0359,  0.0032, -0.0040,  ..., -0.0185, -0.0122, -0.0376]]),\n",
       " 927: tensor([[-0.0144, -0.0122, -0.0068,  ..., -0.0392, -0.0121, -0.0184]]),\n",
       " 928: tensor([[-0.0280,  0.0026, -0.0184,  ..., -0.0332, -0.0162, -0.0470]]),\n",
       " 929: tensor([[-0.0352,  0.0160, -0.0228,  ..., -0.0243, -0.0006, -0.0364]]),\n",
       " 930: tensor([[-0.0429, -0.0085, -0.0129,  ..., -0.0069, -0.0026, -0.0162]]),\n",
       " 931: tensor([[-0.0213, -0.0031, -0.0136,  ..., -0.0356, -0.0195, -0.0326]]),\n",
       " 932: tensor([[-0.0315, -0.0026, -0.0128,  ..., -0.0334,  0.0087, -0.0383]]),\n",
       " 933: tensor([[-0.0215,  0.0006,  0.0045,  ..., -0.0271, -0.0219, -0.0329]]),\n",
       " 934: tensor([[-0.0405, -0.0061, -0.0303,  ..., -0.0334, -0.0094, -0.0200]]),\n",
       " 935: tensor([[-0.0274,  0.0086, -0.0195,  ..., -0.0249, -0.0047, -0.0305]]),\n",
       " 936: tensor([[-0.0541, -0.0052, -0.0207,  ..., -0.0286, -0.0179, -0.0119]]),\n",
       " 937: tensor([[-0.0366, -0.0108,  0.0139,  ..., -0.0180, -0.0163, -0.0204]]),\n",
       " 938: tensor([[-0.0467, -0.0023, -0.0245,  ..., -0.0367, -0.0138, -0.0075]]),\n",
       " 939: tensor([[-0.0272, -0.0013, -0.0188,  ..., -0.0371, -0.0281, -0.0090]]),\n",
       " 940: tensor([[-0.0278,  0.0015, -0.0060,  ..., -0.0139, -0.0101, -0.0278]]),\n",
       " 941: tensor([[-0.0339, -0.0017, -0.0160,  ..., -0.0212, -0.0085, -0.0334]]),\n",
       " 942: tensor([[-0.0419, -0.0082, -0.0138,  ..., -0.0364, -0.0016, -0.0250]]),\n",
       " 943: tensor([[-0.0456, -0.0089, -0.0196,  ..., -0.0378,  0.0123, -0.0184]]),\n",
       " 944: tensor([[-0.0334,  0.0061, -0.0170,  ..., -0.0286,  0.0001, -0.0167]]),\n",
       " 945: tensor([[-0.0345, -0.0007, -0.0193,  ..., -0.0285, -0.0119, -0.0412]]),\n",
       " 946: tensor([[-0.0295, -0.0079, -0.0212,  ..., -0.0243, -0.0033, -0.0374]]),\n",
       " 947: tensor([[-0.0380,  0.0007, -0.0124,  ..., -0.0238,  0.0079, -0.0174]]),\n",
       " 948: tensor([[-0.0275, -0.0012, -0.0209,  ..., -0.0281, -0.0130, -0.0214]]),\n",
       " 949: tensor([[-0.0264,  0.0016, -0.0086,  ..., -0.0178, -0.0099, -0.0290]]),\n",
       " 950: tensor([[-0.0297, -0.0066,  0.0037,  ..., -0.0170, -0.0180, -0.0377]]),\n",
       " 951: tensor([[-0.0411, -0.0086,  0.0158,  ..., -0.0076, -0.0100, -0.0226]]),\n",
       " 952: tensor([[-0.0374,  0.0097, -0.0183,  ..., -0.0457,  0.0111, -0.0127]]),\n",
       " 953: tensor([[-0.0463,  0.0007, -0.0259,  ..., -0.0412, -0.0312, -0.0229]]),\n",
       " 954: tensor([[-0.0405, -0.0117, -0.0246,  ..., -0.0215, -0.0005, -0.0363]]),\n",
       " 955: tensor([[-0.0295,  0.0020, -0.0146,  ..., -0.0480,  0.0018, -0.0142]]),\n",
       " 956: tensor([[-0.0352,  0.0037, -0.0183,  ..., -0.0186, -0.0004, -0.0138]]),\n",
       " 957: tensor([[-0.0321, -0.0005, -0.0290,  ..., -0.0420, -0.0106, -0.0319]]),\n",
       " 958: tensor([[-0.0415, -0.0152, -0.0193,  ...,  0.0069, -0.0005, -0.0281]]),\n",
       " 959: tensor([[-0.0356, -0.0070,  0.0061,  ..., -0.0260, -0.0072, -0.0231]]),\n",
       " 960: tensor([[-0.0304,  0.0050, -0.0333,  ..., -0.0416,  0.0010, -0.0238]]),\n",
       " 961: tensor([[-0.0464,  0.0050, -0.0358,  ..., -0.0452, -0.0066, -0.0262]]),\n",
       " 962: tensor([[-0.0369,  0.0026, -0.0095,  ..., -0.0248,  0.0044, -0.0207]]),\n",
       " 963: tensor([[-0.0384, -0.0014, -0.0281,  ..., -0.0168, -0.0144, -0.0241]]),\n",
       " 964: tensor([[-0.0457,  0.0039, -0.0028,  ..., -0.0308, -0.0366, -0.0233]]),\n",
       " 965: tensor([[-0.0356,  0.0113, -0.0088,  ..., -0.0266,  0.0098, -0.0275]]),\n",
       " 966: tensor([[-0.0194,  0.0058, -0.0113,  ..., -0.0194, -0.0236, -0.0157]]),\n",
       " 967: tensor([[-0.0387,  0.0007, -0.0103,  ..., -0.0303, -0.0103, -0.0172]]),\n",
       " 968: tensor([[-0.0378,  0.0141, -0.0098,  ..., -0.0178, -0.0109, -0.0202]]),\n",
       " 969: tensor([[-0.0405, -0.0141, -0.0112,  ..., -0.0249, -0.0049, -0.0247]]),\n",
       " 970: tensor([[-0.0213, -0.0064,  0.0053,  ..., -0.0264, -0.0028, -0.0264]]),\n",
       " 971: tensor([[-0.0248,  0.0124,  0.0011,  ..., -0.0334, -0.0034, -0.0349]]),\n",
       " 972: tensor([[-0.0134,  0.0054, -0.0033,  ..., -0.0178, -0.0104, -0.0260]]),\n",
       " 973: tensor([[-0.0386, -0.0154, -0.0156,  ..., -0.0442, -0.0046, -0.0242]]),\n",
       " 974: tensor([[-0.0428, -0.0096, -0.0221,  ..., -0.0358,  0.0175, -0.0170]]),\n",
       " 975: tensor([[-0.0312, -0.0053, -0.0058,  ..., -0.0319, -0.0058, -0.0278]]),\n",
       " 976: tensor([[-0.0338, -0.0059, -0.0204,  ..., -0.0249, -0.0030, -0.0400]]),\n",
       " 977: tensor([[-0.0443, -0.0124, -0.0074,  ..., -0.0172,  0.0136, -0.0344]]),\n",
       " 978: tensor([[-0.0458, -0.0057, -0.0071,  ..., -0.0102, -0.0099, -0.0343]]),\n",
       " 979: tensor([[-0.0301, -0.0097, -0.0249,  ..., -0.0211, -0.0132, -0.0238]]),\n",
       " 980: tensor([[-0.0403,  0.0210, -0.0069,  ..., -0.0212, -0.0124, -0.0369]]),\n",
       " 981: tensor([[-0.0305,  0.0132, -0.0163,  ..., -0.0136, -0.0109, -0.0305]]),\n",
       " 982: tensor([[-0.0402, -0.0097, -0.0111,  ..., -0.0374, -0.0121, -0.0387]]),\n",
       " 983: tensor([[-0.0241, -0.0080, -0.0023,  ..., -0.0164, -0.0231, -0.0282]]),\n",
       " 984: tensor([[-0.0275, -0.0067, -0.0300,  ..., -0.0556, -0.0159, -0.0301]]),\n",
       " 985: tensor([[-0.0433,  0.0055, -0.0240,  ..., -0.0274, -0.0166, -0.0194]]),\n",
       " 986: tensor([[-0.0446, -0.0045, -0.0211,  ..., -0.0337,  0.0001, -0.0227]]),\n",
       " 987: tensor([[-0.0523, -0.0033, -0.0202,  ..., -0.0259, -0.0118, -0.0336]]),\n",
       " 988: tensor([[-0.0433, -0.0136, -0.0437,  ..., -0.0216, -0.0142, -0.0136]]),\n",
       " 989: tensor([[-0.0385, -0.0129, -0.0194,  ..., -0.0313, -0.0125, -0.0328]]),\n",
       " 990: tensor([[-0.0309, -0.0018, -0.0063,  ..., -0.0240,  0.0023, -0.0262]]),\n",
       " 991: tensor([[-0.0299,  0.0072, -0.0134,  ..., -0.0450,  0.0005, -0.0149]]),\n",
       " 992: tensor([[-0.0477,  0.0002, -0.0149,  ..., -0.0422, -0.0194, -0.0253]]),\n",
       " 993: tensor([[-0.0403, -0.0138, -0.0082,  ..., -0.0457, -0.0131, -0.0213]]),\n",
       " 994: tensor([[-0.0391,  0.0010, -0.0166,  ..., -0.0278, -0.0307, -0.0208]]),\n",
       " 995: tensor([[-0.0420,  0.0053, -0.0148,  ..., -0.0359,  0.0001, -0.0196]]),\n",
       " 996: tensor([[-0.0391, -0.0082, -0.0006,  ..., -0.0450, -0.0105, -0.0190]]),\n",
       " 997: tensor([[-0.0217, -0.0087, -0.0246,  ..., -0.0317, -0.0149, -0.0266]]),\n",
       " 998: tensor([[-0.0625, -0.0010, -0.0158,  ..., -0.0208, -0.0058, -0.0298]]),\n",
       " 999: tensor([[-0.0303,  0.0024,  0.0018,  ..., -0.0095, -0.0071, -0.0227]]),\n",
       " ...}"
      ]
     },
     "execution_count": 10,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Check node embeddings\n",
    "node_embeddings"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "(129375, torch.Size([1, 1536]))"
      ]
     },
     "execution_count": 15,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Check length and dimension of node embeddings\n",
    "len(node_embeddings), node_embeddings[0].shape"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Check StarQA-PrimeKG splits"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "StarkQA-PrimeKG splits contain train, validation, and test indices for benchmarking the QA-driven AI models."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "dict_keys(['train', 'val', 'test', 'test-0.1'])"
      ]
     },
     "execution_count": 12,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Check the split indices of the starkqa primekg dataframe\n",
    "starkqa_split_indices.keys()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Finally, we can check the number of each split as follows."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "train: 6162\n",
      "val: 2241\n",
      "test: 2801\n",
      "test-0.1: 280\n"
     ]
    }
   ],
   "source": [
    "# Check the number of samples in each split of the starkqa primekg dataframe\n",
    "for split, idx in starkqa_split_indices.items():\n",
    "    print(f\"{split}: {len(idx)}\")"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "venv",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.12.8"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}